torch.roll

张量滚动

1 torch.roll(input, shifts, dims=None) → Tensor

input为输入张量
shift表示要滚动的幅度。负数表示左、上方向（二维时），正数表示右、下方向
dim表示在哪些维度上进行滚动。
如果dim给值，dim的len和shifts的len必须是一样的。比如dims=(a,b)），相当于在第a维度移x位，在b维度上移y位。

【torch】torch.roll相关推荐

【pytorch】——torch.hub
[pytorch]--torch.hub Facebook官方博客表示,PyTorch Hub是一个简易API和工作流程,为复现研究提供了基本构建模块,包含预训练模型库.并且,PyTorch Hub还 ...
【PyTorch】torch.utils.data.Dataset 介绍与实战
文章目录一.前言二.torch.utils.data.Dataset 是什么 1. 干什么用的? 2. 长什么样子? 三.通过继承 torch.utils.data.Dataset 定义自己的数据 ...
【Pytorch】torch.reshape函数
[Pytorch]torch.reshape torch.reshape(input, shape) 返回一个张量,其与input的元素和元素个数相同. >>> a = torch. ...
【PyTorch】torch.cat==＞张量拼接，在图像的应用上可以有效利用原始图像结构信息
1. 字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat 是 concatenate 的意思,即拼接,联系在一起. 使用torch.cat((A,B),dim)时,除拼接维数d ...
【Pytorch】torch.argmax 函数详解
文章目录一.一个参数时的 torch.argmax 函数 1. 介绍 2. 实例二.多个参数时的 torch.argmax 函数 1. 介绍 2. 实例实例1:二维矩阵实例2:三维矩阵实例3 ...
【pytorch】torch.mm，torch.bmm以及torch.matmul的使用
torch.mm torch.mm是两个矩阵相乘,即两个二维的张量相乘如下面的例子 mat1 = torch.randn(2,3) print("mat1=", mat1)mat ...
【PyTorch】torch.nn.Transformer解读与应用
nn.TransformerEncoderLayer 这个类是transformer encoder的组成部分,代表encoder的一个层,而encoder就是将transformerEncoderL ...
【PyTorch】 torch.flatten()与nn.Flatten()的区别
问题 torch.flatten()与nn.Flatten()都可以实现展开Tensor,那么二者的区别是什么呢? 方法经过查阅相关资料,发现二者主要区别有: (1) 默认的dim不同,torch. ...
【pytorch】torch.nn.functional.pad的使用
torch.nn.functional.pad 是对Tensor做padding,输入的参数必须的torch的Tensor 一般地,习惯上会做如下声明 import torch.nn.function ...

【torch】torch.roll

torch.roll

【torch】torch.roll相关推荐

最新文章

热门文章