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)