dim (int) – 索引的轴; index (LongTensor) – 聚合元素的下标; out (Tensor, optional) – 目标张量. torch.index_select.
確定! 回上一頁