tensorly.tenalg.contract¶

contract(tensor1, modes1, tensor2, modes2)[source]¶

Tensor contraction between two tensors on specified modes

Parameters

tensor1tl.tensor

modes1int list or int

modes on which to contract tensor1

tensor2tl.tensor

modes2int list or int

modes on which to contract tensor2

Returns

contractiontensor1 contracted with tensor2 on the specified modes