expand_as
根据 y 的形状扩展 x,扩展后,x 的形状和 y 的形状相同。
x 的维数和 y 的维数应小于等于 6,并且 y 的维数应该大于等于 x 的维数。扩展的维度的维度值应该为 1。
下图展示了一个一维张量通过 expand_as 操作转变为形状为[2,3]的张量。目标张量的形状为[2,3],通过 expand_as,一维张量扩展成形状为[2,3]的张量。
备注
别名支持: 参数名 input 可替代 x, other 可替代 y。
参数
x (Tensor) - 输入的 Tensor,数据类型为:bool、float32、float64、int32 或 int64。 别名:
inputy (Tensor) - 给定输入
x扩展后的形状。 别名:othername (str,可选) - 具体用法请参见 api_guide_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(cpu), stop_gradient=True,
[[1, 2, 3],
[1, 2, 3]])