Skip to main content

dot

Function dot 

Source
pub fn dot<T: TTScalar + Scalar + Default + EinsumScalar>(
    a: &TensorTrain<T>,
    b: &TensorTrain<T>,
) -> Result<T>
Expand description

Free-function wrapper for TensorTrain::dot.

ยงExamples

use tensor4all_simplett::{TensorTrain, AbstractTensorTrain, contraction::dot};

let a = TensorTrain::<f64>::constant(&[2, 3], 3.0);
let b = TensorTrain::<f64>::constant(&[2, 3], 4.0);
let d = dot(&a, &b).unwrap();
// 3*4 * 2*3 = 72
assert!((d - 72.0).abs() < 1e-10);