PyTorch 中的多维张量积

Question

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

Answer 1

如果卷积实际上是指收缩之类的东西，那么您可能正在寻找 torch.tensordot。您可以指定应收缩的指数。