我熟悉einsumNumPy中的工作方式。PyTorch也提供了类似的功能:torch.einsum()。在功能或性能上有何异同?PyTorch文档中提供的信息很少,并且没有提供有关此方面的任何见解。
einsum
python numpy numpy-einsum pytorch pytorch-einsum
numpy ×1
numpy-einsum ×1
python ×1
pytorch ×1
pytorch-einsum ×1