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