pub fn einsum<B: TensorNetworkOps, const D: usize>(
subscripts: &str,
inputs: Vec<Tensor<B, D>>,
) -> Tensor<B, D>Expand description
High-level infallible einsum convenience wrapper.
§Examples
ⓘ
use burn::backend::NdArray;
use burn::tensor::Tensor;
use tenferro_ext_burn::einsum;
let a: Tensor<NdArray<f64>, 2> = Tensor::ones([2, 2], &Default::default());
let b: Tensor<NdArray<f64>, 2> = Tensor::ones([2, 2], &Default::default());
let c = einsum("ij,jk->ik", vec![a, b]);
assert_eq!(c.dims(), [2, 2]);