PyTorch中的多维张量积
在pytorch中，我必须使用[K，L，M]和[M，L，N]维的张量。我想对这些张量沿中间二维进行标准的张量卷积，得到[K，N]张量。我找不到关于如何执行这些操作的官方文档，也许应该在其他库中更好地完成，然后重新转换为pytorch张量？如果卷积实际上是指收缩之类的东西，那么您可能正在寻找。您可以指定应收缩的索引PyTorch中的多维张量积,pytorch,multiplication,tensor,Pytorch,Multiplication,Tensor,在pytorch中，我必须使用[K，L，M]和[M，L，N]维的张量。我想对这些张量沿中间二维进行标准的张量卷积，得到[K，N]张量。我找不到关于如何执行这些操作的官方文档，也许应该在其他库中更好地完成，然后重新转换为pytorch张量？如果卷积实际上是指收缩之类的东西，那么您可能正在寻找。您可以指定应收缩的索引