Python PyTorch atleast_2d用法及代码示例

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

用法:

torch.atleast_2d(*tensors)

参数：

input(Tensor或者张量列表) -

返回：

输出(张量或张量元组)

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

例子：

>>> x = torch.tensor(1.)
>>> x
tensor(1.)
>>> torch.atleast_2d(x)
tensor([[1.]])
>>> x = torch.randn(2,2)
>>> x
tensor([[2.2086, 2.5165],
        [0.1757, 0.5194]])
>>> torch.atleast_2d(x)
tensor([[2.2086, 2.5165],
        [0.1757, 0.5194]])
>>> x = torch.tensor(0.5)
>>> y = torch.tensor(1.)
>>> torch.atleast_2d((x,y))
(tensor([[0.5000]]), tensor([[1.]]))