a = t.arange(0,16).view(4, 4)
a
#选取两个对角线上的元素
index = t.LongTensor([[0, 1, 2, 3], [3, 2, 1, 0]]).t()
b = a.gather(1, index)
b
c = t.zeros(4, 4)
c.scatter_(1,index,b)
报如题所示错误。
解决方法,转换c的类型即可。
c = c.long()
a = t.arange(0,16).view(4, 4)
a
#选取两个对角线上的元素
index = t.LongTensor([[0, 1, 2, 3], [3, 2, 1, 0]]).t()
b = a.gather(1, index)
b
c = t.zeros(4, 4)
c.scatter_(1,index,b)
报如题所示错误。
解决方法,转换c的类型即可。
c = c.long()