tf.broadcast_static_shape

Вычисляет форму результата широковещательного распространения по известным формам.

tf.broadcast_static_shape(
    shape_x, shape_y
)

Если shape_x и shape_y являются полностью известными TensorShapeами, эта функция вычисляет 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])