torch.as_strided¶
- torch.as_strided(input, size, stride, storage_offset=None) Tensor ¶
根据指定的
size
、stride
和storage_offset
创建现有 torch.Tensorinput
的视图。警告
相比于使用 as_strided 手动设置视图的步长,更推荐使用其他视图函数,如
torch.Tensor.expand()
,因为此函数的行为取决于张量存储的实现。构建的存储视图只能引用存储中的元素,否则将抛出运行时错误;如果视图是“重叠的”(多个索引引用内存中同一元素),则其行为是未定义的。- 参数
示例
>>> x = torch.randn(3, 3) >>> x tensor([[ 0.9039, 0.6291, 1.0795], [ 0.1586, 2.1939, -0.4900], [-0.1909, -0.7503, 1.9355]]) >>> t = torch.as_strided(x, (2, 2), (1, 2)) >>> t tensor([[0.9039, 1.0795], [0.6291, 0.1586]]) >>> t = torch.as_strided(x, (2, 2), (1, 2), 1) tensor([[0.6291, 0.1586], [1.0795, 2.1939]])