Как я могу обрезать / удалить часть Тензорного, чтобы соответствовать форме другого Тензорного с 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])