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.]]]))