torch.select

torch.select(input, dim, index) → Tensor

沿给定维度在给定索引处对input张量进行切片。此函数返回原始张量的一个视图，其中删除了给定维度。

注意

如果input是稀疏张量且无法返回张量的视图，则会引发 RuntimeError 异常。在这种情况下，请考虑使用torch.select_copy()函数。

参数

• input (Tensor) – 输入张量。

• dim (int) – 要切片的维度

• index (int) – 要选择的索引

注意

select()等效于切片。例如，tensor.select(0, index)等效于tensor[index]，而tensor.select(2, index)等效于tensor[:,:,index]。