当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch flip用法及代码示例


本文简要介绍python语言中 torch.flip 的用法。

用法:

torch.flip(input, dims) → Tensor

参数

  • input(Tensor) -输入张量。

  • dims(一个列表或者tuple) -要翻转的轴

沿给定轴反转 n-D 张量的顺序,以暗淡为单位。

注意

torch.flip 复制 input 的数据。这与 NumPy 的 np.flip 不同,后者以恒定时间返回视图。由于复制张量的数据比查看该数据需要更多的工作,因此 torch.flip 预计会比 np.flip 慢。

例子:

>>> x = torch.arange(8).view(2, 2, 2)
>>> x
tensor([[[ 0,  1],
         [ 2,  3]],

        [[ 4,  5],
         [ 6,  7]]])
>>> torch.flip(x, [0, 1])
tensor([[[ 6,  7],
         [ 4,  5]],

        [[ 2,  3],
         [ 0,  1]]])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.flip。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。