tf.broadcast_dynamic_shape
Вычисляет форму широковещательной рассылки, заданную символическими формами.
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/api_docs/python/tf/broadcast_dynamic_shape