2020年5月14日
gather函数
摘要： gather(input, dim, index)：根据 index，在 dim 维度上选取数据，输出的 size 与 index 一致 # input (Tensor) – 源张量 # dim (int) – 索引的轴 # index (LongTensor) – 聚合元素的下标(index需要是 阅读全文
posted @ 2020-05-14 12:59 珠江水手 阅读(2160) 评论(0) 推荐(0) 编辑