Python PyTorch atleast_1d用法及代码示例

本文简要介绍python语言中 torch.atleast_1d 的用法。

用法:

torch.atleast_1d(*tensors)

参数：

input(Tensor或者张量列表) -

返回：

输出(张量或张量元组)

返回每个零维度输入张量的一维视图。具有一维或多维的输入张量按原样返回。

例子：

>>> x = torch.randn(2)
>>> x
tensor([1.4584, 0.7583])
>>> torch.atleast_1d(x)
tensor([1.4584, 0.7583])
>>> 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.]))