broadcast_shape¶
该函数返回对 x_shape 大小的张量和 y_shape 大小的张量做 broadcast 操作后得到的 shape,如您想了解更多,请参见 cn_user_guide_broadcasting 。
参数¶
x_shape (list[int]|tuple[int]) - 输入 Tensor 的 shape。
y_shape (list[int]|tuple[int]) - 输入 Tensor 的 shape。
返回¶
broadcast 操作后的 shape,返回类型为 list[int]。
代码示例¶
import paddle
shape = paddle.broadcast_shape([2, 1, 3], [1, 3, 1])
# [2, 3, 3]
# shape = paddle.broadcast_shape([2, 1, 3], [3, 3, 1])
# ValueError (terminated with error message).