torch.atleast_3d¶
- torch.atleast_3d(*tensors)[source]¶
返回每个输入张量的 3 维视图,其维数为零。具有三个或更多维的输入张量将按原样返回。
示例
>>> x = torch.tensor(0.5) >>> x tensor(0.5000) >>> torch.atleast_3d(x) tensor([[[0.5000]]]) >>> y = torch.arange(4).view(2, 2) >>> y tensor([[0, 1], [2, 3]]) >>> torch.atleast_3d(y) tensor([[[0], [1]], [[2], [3]]]) >>> x = torch.tensor(1).view(1, 1, 1) >>> x tensor([[[1]]]) >>> torch.atleast_3d(x) tensor([[[1]]]) >>> x = torch.tensor(0.5) >>> y = torch.tensor(1.0) >>> torch.atleast_3d((x, y)) (tensor([[[0.5000]]]), tensor([[[1.]]]))