关于torch.bmm的工作原理

torch.bmm(input, mat2, deterministic=False, out=None) → Tensor Performs a batch matrix-matrix product of matrices stored in input and mat2. input and mat2 must be 3-D tensors each containing the same

>>阅读原文<<