torch.gather(input, dim, index, out=None) → Tensor. input (Tensor) – 源张量 dim (int) – 索引的轴 index (LongTensor) – 聚合元素的下标
確定! 回上一頁