PyTorch 中的多维张量积

Question

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

Answer 1

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