tensorly.tenalg
batched_tensor_dot
Returns a generalized outer product of the two tensors
of shape (n_samples, J1, …, JN)
of shape (n_samples, K1, …, KM)
of shape (n_samples, J1, …, JN, K1, …, KM)