Function autograd::ops::batch_matmul

pub fn batch_matmul(a: &Tensor, b: &Tensor) -> Tensor

Batched matrix multiplication.

The rank of a and b must be equals.

extern crate autograd as ag;

let mut ctx = ag::Context::new();
let ref a = ag::zeros(&[2, 3, 4, 2]);
let ref b = ag::zeros(&[2, 3, 2, 3]);
let ref c = ag::batch_matmul(a, b);

assert_eq!(c.eval(&mut ag::Context::new()).shape(), &[2, 3, 4, 3]);

For detailed description, see https://www.tensorflow.org/api_docs/python/tf/matmul