Python tf.broadcast_static_shape用法及代码示例

在给定已知形状的情况下计算广播的形状。

用法

tf.broadcast_static_shape(
    shape_x, shape_y
)

当 shape_x 和 shape_y 是完全已知的 TensorShape s 时，这会计算 TensorShape，它是在形状 shape_x 和 shape_y 的张量中应用的广播操作的结果的形状。

例如，如果 shape_x 是 TensorShape([1, 2, 3]) 并且 shape_y 是 TensorShape([5, 1, 3]) ，则结果是一个 TensorShape ，其值为 TensorShape([5, 2, 3]) 。

当张量具有静态已知形状时，这在验证广播操作的结果时很有用。

例子：

shape_x = tf.TensorShape([1, 2, 3])
shape_y = tf.TensorShape([5, 1 ,3])
tf.broadcast_static_shape(shape_x, shape_y)
TensorShape([5, 2, 3])