tf.broadcast_static_shape

Вычисляет форму расчётного результата, задавая известные формы.

tf.broadcast_static_shape(
    shape_x, shape_y
)

Когда shape_x и shape_y полностью известны TensorShapes это вычисляет TensorShape, которая является формой результата операции broadcast, применённой к тензорам с формами shape_x и shape_y.

Например, если shape_x равно TensorShape([1, 2, 3]) и shape_y равно TensorShape([5, 1, 3]), результат — TensorShape со значением TensorShape([5, 2, 3]).

Это полезно для проверки результата операции broadcast, когда тензоры имеют статически известные формы.

Пример:

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