expand

paddle. expand ( x: Tensor, shape: ShapeLike, name: str | None = None ) → Tensor [source]

Expand the input tensor to a given shape.

Both the number of dimensions of x and the number of elements in shape should be less than or equal to 6. And the number of dimensions of x should be less than the number of elements in shape. The dimension to expand must have a value 0.

The image illustrates a typical case of the expand operation. The Original Tensor is a 1D tensor with shape [3] and values [1, 2, 3]. Using the paddle.expand method with the parameter shape = [2, 3], it is broadcasted and expanded into a 2D tensor with shape [2, 3]

legend of expand API

Note

Alias Support: The parameter name input can be used as an alias for x and size can be used as an alias for shape. shape can be a variable number of arguments. For example:

System Message: ERROR/3 (/usr/local/lib/python3.10/site-packages/paddle/tensor/manipulation.py:docstring of paddle.tensor.manipulation.expand, line 17)

Unexpected indentation.

paddle.expand(tensor_x, shape=[3, 4], name=None) tensor_x.expand([3, 4]) -> paddle.expand(tensor_x, [3, 4]) tensor_x.expand(3, 4) -> paddle.expand(tensor_x, 3, 4) tensor_x.expand(size=[3, 4]) -> paddle.expand(tensor_x, size=[3, 4])

Parameters

  • x (Tensor) – The input Tensor, its data type is bool, float16, float32, float64, int32, int64, uint8, uint16, complex64 or complex128. alias: input

  • shape (list|tuple|Tensor|variable number of arguments) – The result shape after expanding. The data type is int32. If shape is a list or tuple, all its elements should be integers or 0-D or 1-D Tensors with the data type int32. If shape is a Tensor, it should be an 1-D Tensor with the data type int32. The value -1 in shape means keeping the corresponding dimension unchanged. shape can be a variable number of arguments. alias: size.

  • name (str|None, optional) – The default value is None. Normally there is no need for user to set this property. For more information, please refer to api_guide_Name .

Returns

N-D Tensor, A Tensor with the given shape. The data type is the same as x.

Examples

>>> import paddle

>>> data = paddle.to_tensor([1, 2, 3], dtype='int32')
>>> out = paddle.expand(data, shape=[2, 3])
>>> print(out)
Tensor(shape=[2, 3], dtype=int32, place=Place(cpu), stop_gradient=True,
[[1, 2, 3],
 [1, 2, 3]])