tf.broadcast_dynamic_shape
| Просмотреть исходный код на GitHub |
Вычисляет форму широковещательной передачи, используя символические формы.
tf.broadcast_dynamic_shape(
shape_x, shape_y
)
Когда shape_x и shape_y являются тензорами, представляющими формы (т.е. результатом вызова tf.shape другого тензора), это вычисляет тензор, который представляет форму результата операции широковещательной передачи, применяемой к тензорам с формами shape_x и shape_y.
Это полезно для проверки результата операции широковещательной передачи, когда тензоры не имеют статически известных форм.
Пример:
shape_x = (1, 2, 3) shape_y = (5, 1, 3) tf.broadcast_dynamic_shape(shape_x, shape_y) <tf.Tensor: shape=(3,), dtype=int32, numpy=array([5, 2, 3], ...>
| Аргументы | |
|---|---|
shape_x | Целочисленный вектор ранга 1 Tensor, представляющий форму x. |
shape_y | Целочисленный вектор ранга 1 Tensor, представляющий форму y. |
| Возвращаемое значение | |
|---|---|
Целочисленный вектор ранга 1 Tensor , представляющий форму широковещательной передачи. |
| Возбуждает исключения | |
|---|---|
InvalidArgumentError | Если две формы несовместимы для широковещательной передачи. |
© 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/r2.4/api_docs/python/tf/broadcast_dynamic_shape