Python PyTorch atleast_3d用法及代码示例

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

用法:

torch.atleast_3d(*tensors)

参数：

input(Tensor或者张量列表) -

返回：

输出(张量或张量元组)

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

示例

>>> x = torch.tensor(0.5)
>>> x
tensor(0.5000)
>>> torch.atleast_3d(x)
tensor([[[0.5000]]])
>>> y = torch.randn(2,2)
>>> y
tensor([[-0.8079,  0.7460],
        [-1.1647,  1.4734]])
>>> torch.atleast_3d(y)
tensor([[[-0.8079],
        [ 0.7460]],

        [[-1.1647],
        [ 1.4734]]])
>>> x = torch.randn(1,1,1)
>>> x
tensor([[[-1.5689]]])
>>> torch.atleast_3d(x)
tensor([[[-1.5689]]])
>>> x = torch.tensor(0.5)
>>> y = torch.tensor(1.)
>>> torch.atleast_3d((x,y))
(tensor([[[0.5000]]]), tensor([[[1.]]]))