tenferro_to_burn

Function tenferro_to_burn 

Source
pub fn tenferro_to_burn<B: Backend<FloatElem = f64>>(
    tensor: Tensor<f64>,
    device: &B::Device,
) -> FloatTensor<B>
Expand description

Convert a tenferro Tensor<f64> into a Burn backend tensor primitive, panicking if conversion fails.

§Examples

use burn::backend::NdArray;
use tenferro_ext_burn::convert::tenferro_to_burn;

let device = Default::default();
let tenferro_t: tenferro_tensor::Tensor<f64> = todo!();
let _burn_prim = tenferro_to_burn::<NdArray<f64>>(tenferro_t, &device);