2020年6月14日
gather
摘要： 1、作用是将input中的数据按dim和index取出2、out[i][j][k] = input[index[i][j][k]][j][k] # if dim == 0 out[i][j][k] = input[i][index[i][j][k]][k] # if dim == 1out[i][j 阅读全文
posted @ 2020-06-14 15:13 6+0 阅读(267) 评论(0) 推荐(0) 编辑