expand_as¶
根据 y
的形状扩展 x
,扩展后,x
的形状和 y
的形状相同。
x
的维数和 y
的维数应小于等于 6,并且 y
的维数应该大于等于 x
的维数。扩展的维度的维度值应该为 1。
参数¶
x (Tensor) - 输入的 Tensor,数据类型为:bool、float32、float64、int32 或 int64。
y (Tensor) - 给定输入
x
扩展后的形状。name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor,数据类型与 x
相同。
代码示例¶
import paddle
data_x = paddle.to_tensor([1, 2, 3], 'int32')
data_y = paddle.to_tensor([[1, 2, 3], [4, 5, 6]], 'int32')
out = paddle.expand_as(data_x, data_y)
print(out)
# Tensor(shape=[2, 3], dtype=int32, place=Place(gpu:0), stop_gradient=True,
# [[1, 2, 3],
# [1, 2, 3]])