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)