torch.roll

原理：将tensor进行偏移。例如[1,2,3]变成[2,3,1]，这个就是roll。又再如[1,2,3]变成[3,1,2]。前者向左偏移一个单位，后者向左偏移两个单位。

明白了原理就很简单了。

torch.roll(input, shifts, dims=None)

实战

import torch
x = torch.tensor([[1, 2],[ 3,4]])

print(x)
torch.roll(x,1,-1)#表示最后一个维度上进行操作，并且向右偏移1个单位。

print(x)
torch.roll(x,-1,0)#表示在0维度上进行操作，并且向上偏移1个单位。

print(x)
torch.roll(x,(2,2),(0,1))#表示两个维度上都要进行偏移，并且各自在正方向上偏移两个单位。

torch.roll相关推荐

torch.roll() 详解
torch.roll(input, shifts, dims=None) input (Tensor) – the input tensor. shifts (int or tuple of pyth ...
【torch】torch.roll
torch.roll 张量滚动 1 torch.roll(input, shifts, dims=None) → Tensor input为输入张量 shift表示要滚动的幅度.负数表示左.上方向(二 ...
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(input, shifts, dims=None) -> Tensor 定义沿给定维度滚动张量输入.超出最后一个位置的元素在第一个位置重新引入.如果 dims 为 Non ...
[pytorch]torch.roll函数
torch中的roll函数可以用于张量的位置变换操作. 博客推荐 import torch import numpy as np import matplotlib.pyplot as pltshif ...
torch.roll() 函数用法
Pytorch 官方文档:https://pytorch.org/docs/master/generated/torch.roll.htmlhttps://pytorch.org/docs/maste ...
torch(七)、Math operations(2)
参考 torch(七).Math operations(2) - 云+社区 - 腾讯云目录 Spectral Ops torch.fft(input, signal_ndim, normalized ...
Swin Transformer对CNN的降维打击
一.前言一张图告诉你Transformer现在是多么的强!几乎包揽了ADE20K语义分割的前几名! 该文章详细解读Swin-transformer的相关内容以及高明之处.看完学不会,你在评论区打我! ...

torch.roll

torch.roll相关推荐

最新文章

热门文章