我们以下面的例子说明,可以直接使用flip()函数实现行数据的互换。
import torch
a = torch.rand(2, 5)
print(a)
b = a.flip(0)
print(b)
结果如下:
tensor([[0.4037, 0.1549, 0.4080, 0.3528, 0.3121],
[0.2444, 0.5353, 0.9513, 0.9481, 0.1902]])
tensor([[0.2444, 0.5353, 0.9513, 0.9481, 0.1902],
[0.4037, 0.1549, 0.4080, 0.3528, 0.3121]])