文章目录 torch.roll()的官方文档 函数形式及参数说明 举个例子 torch.roll()的应用 具体步骤 torch.roll()的官方文档 函数形式及参数说明 函数形式:torch.roll(input,shifts,dims=None) 参数解释: input:输入张量 shifts:滚动的方向和长度,若为正,则向下滚动;若为负,则向上滚动。可是一个整数也可以是一个元组。(具体见下面例子) dims:张量按什么维度滚动。以二维为例,0就是按行上下或者下上(shifts为负)滚动;1就是左右或者右左(shifts为负)。可以是整数或元组。 </