tensorly.tenalg.tensordot

tensordot(tensor1, tensor2, modes, batched_modes=())[source]

Batched tensor contraction between two tensors on specified modes

Parameters:

tensor1tl.tensor

tensor2tl.tensor

modesint list or int

modes on which to contract tensor1 and tensor2

batched_modesint or tuple[int]

Returns:

contractiontensor1 contracted with tensor2 on the specified modes