// Fork me on GitHub
2020年8月26日
pytorch 函数
摘要： x=torch.tensor([[1,2,3], [4,5,6]]) torch.gather(dim=0,index=torch.tensor([0,0,1],[1,0,1])) 表示提取索引元素，索引由index提供的信息生成）（索引！=index） dim=0,index里的元素表示 索引的第 阅读全文
posted @ 2020-08-26 09:47 繁星春水 阅读(115) 评论(0) 推荐(0) 编辑
1 //2 3
//4