样本张量:
tensor([[ 0., 1., 2., 3., 4., 5.], # class1
[ 6., 7., 8., 9., 10., 11.], # class3
[12., 13., 14., 15., 16., 17.], # class2
[18., 19., 20., 21., 22., 23.], # class0
[24., 25., 26., 27., 28., 29.]. # class1
])
预期结果:
tensor([[18., 19., 20., 21., 22., 23.], # class0
[12., 13., 14., 15., 16., 17.], # class1
[12., 13., 14., 15., 16., 17.], # class2
[ 6., 7., 8., 9., 10., 11.]. # class3
])
是否有一个纯PyTorch方法来实现这一点
您可以使用^{} 根据类索引添加,然后除以每个标签的编号,使用^{} 计算:
正如预期的那样
result
:相关问题 更多 >
编程相关推荐