tf.broadcast_static_shape
| Просмотреть исходный код на GitHub |
Вычисляет форму широковещательной передачи, зная формы.
tf.broadcast_static_shape(
shape_x, shape_y
)
Когда shape_x и shape_y являются полностью известными TensorShapes, это вычисляет TensorShape, которая является формой результата операции широковещательной передачи, применяемой к тензорам с формами shape_x и shape_y.
Например, если shape_x — [1, 2, 3], а shape_y — [5, 1, 3], результат — TensorShape со значением [5, 2, 3].
Это полезно при проверке результата операции широковещательной передачи, когда тензоры имеют статически известные формы.
| Аргументы | |
|---|---|
shape_x | A TensorShape |
shape_y | A TensorShape |
| Возвращаемое значение | |
|---|---|
A TensorShape представляющая широковещательную форму. |
| Исключения | |
|---|---|
ValueError | Если две формы не могут быть переданы в широковещательной форме. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/broadcast_static_shape