如何在不知道维度的情况下拼接张量？

Question

假设我有一个形状为

x

的张量

(21, 256, *)

，我怎样才能以编程方式获得相当于

x[:, :48, ...]

的张量？

Answer 1

上述问题的等价物是使用

torch.index_select

：

indexes = torch.arange(48)
torch.index_select(myTensor, 1, indexes)