tensorly.tenalg.batched_tensor_dot

batched_tensor_dot(tensor1, tensor2)[source]

Returns a generalized outer product of the two tensors

Parameters:
tensor1tensor

of shape (n_samples, J1, …, JN)

tensor2tensor

of shape (n_samples, K1, …, KM)

Returns:
outer product of tensor1 and tensor2

of shape (n_samples, J1, …, JN, K1, …, KM)