torch.flip()

直接案例

import torch
x = torch.arange(24).view(2, 3, 4)
x
tensor([[[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]],

[[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23]]])



torch.flip(x, [0])
tensor([[[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23]],

[[ 0,  1,  2,  3],
 [ 4,  5,  6,  7],
 [ 8,  9, 10, 11]]])


torch.flip(x, [1])
tensor([[[ 8, 9, 10, 11],
[ 4, 5, 6, 7],
[ 0, 1, 2, 3]],

[[20, 21, 22, 23],
 [16, 17, 18, 19],
 [12, 13, 14, 15]]]),




torch.flip(x, [0, 1])
tensor([[[20, 21, 22, 23],
[16, 17, 18, 19],
[12, 13, 14, 15]],

[[ 8,  9, 10, 11],
 [ 4,  5,  6,  7],
 [ 0,  1,  2,  3]]])


```