torch.Tensor.take_along_dim¶ Tensor.take_along_dim(indices, dim) → Tensor¶ 请参阅 torch.take_along_dim()