pytorch之torch.flip

torch.flip(input,dim):第一个参数是输入，第二个参数是输入的第几维度，按照维度对输入进行翻转

import torch
x = torch.arange(16).view(2, 2, 2,2)
print('x=\n',x)
a = torch.flip(x, [2])
print('a=\n',a)

x=tensor([[[[ 0,  1],[ 2,  3]],[[ 4,  5],[ 6,  7]]],[[[ 8,  9],[10, 11]],[[12, 13],[14, 15]]]])
a=tensor([[[[ 2,  3],[ 0,  1]],[[ 6,  7],[ 4,  5]]],[[[10, 11],[ 8,  9]],[[14, 15],[12, 13]]]])Process finished with exit code 0

import torch
x = torch.arange(16).view(2, 2, 2,2)
print('x=\n',x)
a = torch.flip(x, [3])
print('a=\n',a)

x=tensor([[[[ 0,  1],[ 2,  3]],[[ 4,  5],[ 6,  7]]],[[[ 8,  9],[10, 11]],[[12, 13],[14, 15]]]])
a=tensor([[[[ 1,  0],[ 3,  2]],[[ 5,  4],[ 7,  6]]],[[[ 9,  8],[11, 10]],[[13, 12],[15, 14]]]])Process finished with exit code 0

pytorch之torch.flip相关推荐

2021.08.23学习内容Pytorch与Torch的关系以及Torchvision作用
** PyTorch **: 是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序. 2017年1月,由Facebook人工智能研究院(FAIR)基于Torch推出了PyTo ...
8_用opencv调用深度学习框架tenorflow、Pytorch、Torch、caffe训练好的模型（20190212）
用opencv调用深度学习框架tenorflow.Pytorch.Torch.caffe训练好模型(20190212) 文章目录: https://blog.csdn.net/hust_bochu_x ...
Pytorch学习-torch.max()和min()深度解析
Pytorch学习-torch.max和min深度解析 max的使用 min同理 dim参数理解二维张量使用max() 三维张量使用max() max的使用 min同理参考链接: 参考链接: 对于 ...
【PyTorch】torch.cat==＞张量拼接，在图像的应用上可以有效利用原始图像结构信息
1. 字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat 是 concatenate 的意思,即拼接,联系在一起. 使用torch.cat((A,B),dim)时,除拼接维数d ...
Pytorch和Torch和torchvision的安装
超详细的Pytorch和torch和torchvision的安装步骤~~~~~ Pytorch安装步骤包地址:https://download.pytorch.org/whl/torch_stabl ...
YDOOK：Pytorch : AI : torch.tensor.size() 与 torch.tensor.shape 的区别
YDOOK:Pytorch : AI : torch.tensor.size() 与 torch.tensor.shape 的区别区别: 1. torch.tensor.size() 可用通过 :t ...
Pytorch中torch.nn.Softmax的dim参数含义
自己搞了一晚上终于搞明白了,下文说的很透彻,做个记录,方便以后翻阅 Pytorch中torch.nn.Softmax的dim参数含义
深入浅出Pytorch函数——torch.zeros
分类目录:<深入浅出Pytorch函数>总目录相关文章: · 深入浅出Pytorch函数--torch.Tensor · 深入浅出Pytorch函数--torch.ones · 深入浅出 ...
深入浅出Pytorch函数——torch.arange
分类目录:<深入浅出Pytorch函数>总目录相关文章: · 深入浅出TensorFlow2函数--tf.range · 深入浅出Pytorch函数--torch.arange · 深入 ...

pytorch之torch.flip

pytorch之torch.flip相关推荐

最新文章

热门文章