einsum

Function einsum 

Source
pub fn einsum<B: TensorNetworkOps, const D: usize>(
    _subscripts: &str,
    _inputs: Vec<Tensor<B, D>>,
) -> Tensor<B, D>
Expand description

High-level einsum on Burn tensors, dispatching to the backend’s TensorNetworkOps::tn_einsum implementation.

§Examples

use burn::backend::NdArray;
use burn::tensor::Tensor;
use tenferro_burn::einsum;

let a: Tensor<NdArray<f64>, 2> = Tensor::ones([3, 4], &Default::default());
let b: Tensor<NdArray<f64>, 2> = Tensor::ones([4, 5], &Default::default());
let c: Tensor<NdArray<f64>, 2> = einsum("ij,jk->ik", vec![a, b]);