pytorch中index_select()的用法
pytorch中index_select()的用法
刚开始学习pytorch,遇到了index_select(),一开始不太明白几个参数的意思,后来查了一下资料,算是明白了一点。
a = torch.linspace(1, 12, steps=12).view(3, 4)
print(a)
b = torch.index_select(a, 0, torch.tensor([0, 2]))
pr...
原创
2019-11-01 11:33:26 ·
52061 阅读 ·
7 评论