Python PyTorch ravel用法及代碼示例

本文簡要介紹python語言中 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])