torch.reshape¶
- torch.reshape(input, shape) Tensor ¶
返回一个与
input
具有相同数据和元素数量的张量,但具有指定的形状。如果可能,返回的张量将是input
的视图。否则,它将是副本。连续输入和具有兼容步幅的输入可以不复制地重塑,但您不应依赖于复制与查看行为。请参阅
torch.Tensor.view()
,了解何时可以返回视图。单个维度可以是 -1,在这种情况下,它将从剩余维度和
input
中的元素数量推断出来。示例
>>> a = torch.arange(4.) >>> torch.reshape(a, (2, 2)) tensor([[ 0., 1.], [ 2., 3.]]) >>> b = torch.tensor([[0, 1], [2, 3]]) >>> torch.reshape(b, (-1,)) tensor([ 0, 1, 2, 3])