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]);