tf.broadcast_dynamic_shape
| View source on 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 | Если две формы несовместимы для широковещательной рассылки. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/broadcast_dynamic_shape