burn_to_tenferro

Function burn_to_tenferro 

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

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

§Examples

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

let burn_prim: <NdArray<f64> as burn::tensor::backend::Backend>::FloatTensorPrimitive =
    todo!();
let tenferro_t = burn_to_tenferro::<NdArray<f64>>(burn_prim);
assert_eq!(tenferro_t.dims().len(), 1);