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


Python PyTorch Tensor.stride用法及代码示例


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

用法:

Tensor.stride(dim) → tuple or int

参数

dim(int,可选的) -需要步幅的所需尺寸

返回self 张量的步幅。

步幅是在指定维度 dim 中从一个元素到下一个元素所需的跳跃。如果没有传入参数,则返回所有步幅的元组。否则,返回一个整数值作为特定维度 dim 中的步幅。

例子:

>>> x = torch.tensor([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
>>> x.stride()
(5, 1)
>>> x.stride(0)
5
>>> x.stride(-1)
1

相关用法


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