torch.roll() 用法解读

torch.roll(input, shifts, dims=None) -> Tensor

定义

沿给定维度滚动张量输入。超出最后一个位置的元素在第一个位置重新引入。如果 dims 为 None，则张量将在滚动前被展平，然后恢复到原始形状。

参数

input (Tensor) - 输入张量
shifts (int or tuple of python: ints) - 张量元素移动的位置数。如果 shifts 是元组，则 dims 必须是大小相同的元组，每个维度都会滚动对应的值
dims (int or tuple of python: ints) - 沿着dims指定的维度进行滚动

示例

import torchx = torch.tensor([1, 2, 3, 4, 5, 6, 7, 8]).view(4, 2)
print(x)
# tensor([[1, 2],
#         [3, 4],
#         [5, 6],
#         [7, 8]])
x1 = torch.roll(x, 1)
print(x1)
# tensor([[8, 1],
#         [2, 3],
#         [4, 5],
#         [6, 7]])
x2 = torch.roll(x, 1, 0)
print(x2)
# tensor([[7, 8],
#         [1, 2],
#         [3, 4],
#         [5, 6]])
x3 = torch.roll(x, -1, 0)
print(x3)
# tensor([[3, 4],
#         [5, 6],
#         [7, 8],
#         [1, 2]])
x4 = torch.roll(x, shifts=(2, 1), dims=(0, 1))
print(x4)
# tensor([[6, 5],
#         [8, 7],
#         [2, 1],
#         [4, 3]])

参考

torch.roll — PyTorch 1.12 documentation

torch.roll() 用法解读相关推荐

torch.roll() 函数用法
Pytorch 官方文档:https://pytorch.org/docs/master/generated/torch.roll.htmlhttps://pytorch.org/docs/maste ...
torch.roll
原理:将tensor进行偏移.例如[1,2,3]变成[2,3,1],这个就是roll.又再如[1,2,3]变成[3,1,2].前者向左偏移一个单位,后者向左偏移两个单位. 明白了原理就很简单了. to ...
torch.stack(), torch.cat()用法详解
torch.stack(), torch.cat()用法详解 if __name__ == '__main__':import torchx_dat = torch.tensor([[1, 2], [ ...
torch.roll() 详解
torch.roll(input, shifts, dims=None) input (Tensor) – the input tensor. shifts (int or tuple of pyth ...
input标签用法解读
HTML5/HTML中标签用法解读 OK!今天博主为小伙伴们介绍的内容是HTML5/HTML中标签的用法,,, ,emmm图文并茂哦! 下面正式开始内容的介绍:首先,直观上说标签规定了用户可以在其中输 ...
【torch】torch.roll
torch.roll 张量滚动 1 torch.roll(input, shifts, dims=None) → Tensor input为输入张量 shift表示要滚动的幅度.负数表示左.上方向(二 ...
torch.sort()用法
torch.sort()用法 torch.sort(input, dim=- 1, descending=False, stable=False, out=None) 这个函数用于将输入的tensor ...
torch.roll 函数的理解
torch.roll 函数官方解释如果是看swin-transformer进来的,推荐看看GitHub上的这个问题,会很有帮助! https://github.com/microsoft/Swin- ...
torch.roll图片实验
torch.roll(input, shifts, dims=None) → Tensor input为输入张量,shifts表示要滚动的方向.负数表示左上,正数表示右下.dims表示要滚动的维度. ...

torch.roll() 用法解读

定义

参数

示例

参考

torch.roll() 用法解读相关推荐

最新文章

热门文章