2020年11月28日
pytorch index_select()函数
摘要： 函数实现从当前张量中从某个维度选择一部分序号的张量 tensor.select_index(dim, index) 对于一个二维张量feature： 第一个参数 参数0表示按行索引，1表示按列进行索引 第二个参数 是一个整数类型的一维tensor，就是索引的序号 二维张量举例： 三维张量举例： 另一 阅读全文
posted @ 2020-11-28 11:34 爱吃砂糖橘的白龙 阅读(1109) 评论(0) 推荐(0) 编辑