pytorch - PyTorch 中的多维张量积

Question

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

标签： pytorchmultiplicationtensor

Answer 1

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