PyTorch中的多维张量积

Question

在pytorch中，我必须使用尺寸为[K，L，M]和[M，L，N]的张量。我想对这些张量沿中间二维进行标准张量卷积，以获得[K，N]张量。我找不到有关如何执行这些操作的官方文档，也许最好在其他库中完成然后将其转换为pytorch张量？

Answer 1

如果通过卷积实际上意味着诸如收缩之类的东西，那么您可能正在寻找torch.tensordot。您可以指定应收缩的索引。