Flip
torch_flip(self, dims)
self | (Tensor) the input tensor. |
---|---|
dims | (a list or tuple) axis to flip on |
Reverse the order of a n-D tensor along given axis in dims.
#> torch_tensor #> (1,.,.) = #> 6 7 #> 4 5 #> #> (2,.,.) = #> 2 3 #> 0 1 #> [ CPUFloatType{2,2,2} ]