如何使用 PyTorch 修剪/移除张量的一部分以匹配另一个张量的形状？

Question

我有 2 个张量：

outputs: torch.Size([4, 27, 161])       pred: torch.Size([4, 30, 161])

我想pred（从最后）剪切，以便它具有与outputs.

用 PyTorch 做到这一点的最佳方法是什么？

Answer 1

您可以使用窄

例如：

a = torch.randn(4,30,161)
a.size() # torch.Size([4, 30, 161])
a.narrow(1,0,27).size() # torch.Size([4, 27, 161])