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])