torch.atleast_1d¶
- torch.atleast_1d(*tensors)[源代码]¶
返回每个零维输入张量的 1 维视图。具有一个或多个维度的输入张量将按原样返回。
示例
>>> x = torch.arange(2) >>> x tensor([0, 1]) >>> torch.atleast_1d(x) tensor([0, 1]) >>> 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.]))