摘要: 转自:https://blog.csdn.net/tfcy694/article/details/85332953 按照指定轴上的坐标进行过滤 index_select()沿着某tensor的一个轴dim筛选若干个坐标 >>> x = torch.randn(3, 4) # 目标矩阵 >>> x t 阅读全文
posted @ 2021-09-06 19:31 鲍曼小学生 阅读(1818) 评论(0) 推荐(0) 编辑