torch.ravel

torch.ravel(input) → Tensor

返回一个连续的展平张量。只有在必要时才会进行复制。

参数

input (Tensor) – 需要输入的张量。

示例：

>>> t = torch.tensor([[[1, 2],
...                    [3, 4]],
...                   [[5, 6],
...                    [7, 8]]])
>>> torch.ravel(t)
tensor([1, 2, 3, 4, 5, 6, 7, 8])