expand( x, shape, name=None )
Expand the input tensor to a given shape.
Both the number of dimensions of
xand the number of elements in
shapeshould be less than or equal to 6. The dimension to expand must have a value 1.
x (Tensor) – The input tensor, its data type is bool, float32, float64, int32 or int64.
shape (list|tuple|Tensor) – The result shape after expanding. The data type is int32. If shape is a list or tuple, all its elements should be integers 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.
name (str, optional) – The default value is None. Normally there is no need for user to set this property. For more information, please refer to Name .
A Tensor with the given shape. The data type is the same as
- Return type
import paddle data = paddle.to_tensor([1, 2, 3], dtype='int32') out = paddle.expand(data, shape=[2, 3]) print(out) # [[1, 2, 3], [1, 2, 3]]