torch.atleast_1d

torch.atleast_1d(*tensors)

返回每個零維輸入張量的一維檢視。具有一維或更多維度的輸入張量原樣返回。

引數

input (張量 或 張量列表) –

返回

輸出 (張量 或 張量元組)

示例

>>> x = torch.arange(2)
>>> x
tensor([0, 1])
>>> torch.atleast_1d(x)
tensor([0, 1])
>>> x = torch.tensor(1.)
>>> x
tensor(1.)
>>> torch.atleast_1d(x)
tensor([1.])
>>> x = torch.tensor(0.5)
>>> y = torch.tensor(1.)
>>> torch.atleast_1d((x, y))
(tensor([0.5000]), tensor([1.]))