mindspore.ops.atleast_1d
- mindspore.ops.atleast_1d(inputs)[source]
Reshapes Tensor in inputs, every Tensor has at least one dimension after this operation.
Scalar is converted to a 1-D Tensor, input tensor with one or more dimensions will be returned as it is.
- Parameters
inputs (Union[Tensor, list[Tensor]]) – One or more input tensors.
- Returns
Tensor or list[Tensor]. If returned a list, every element a in that list satisfies a.ndim >= 1.
- Raises
TypeError – If the inputs is not a tensor or a list of tensors.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import numpy as np >>> from mindspore import Tensor, ops >>> x1 = Tensor(np.ones((2, 3))) >>> x2 = Tensor(np.ones(())) >>> x3 = Tensor(np.ones(5)) >>> out = ops.atleast_1d([x1, x2, x3]) >>> print(out[0].asnumpy()) [[1. 1. 1.] [1. 1. 1.]] >>> print(out[1].asnumpy()) [1.] >>> print(out[2].asnumpy()) [1. 1. 1. 1. 1.]