2020年4月24日
pytorch-gather
摘要： 函数torch.gather(input, dim, index, out=None, sparse_grad=False) → Tensor沿给定轴 dim ,将输入索引张量 index 指定位置的值进行聚合. input (Tensor) – 原张量 dim (int) – 索引的轴，二维中di 阅读全文
posted @ 2020-04-24 08:50 OliYoung 阅读(514) 评论(0) 推荐(0)