View source on GitHub |
Computes the shape of a broadcast given symbolic shapes.
tf.broadcast_dynamic_shape( shape_x, shape_y )
When shape_x and shape_y are Tensors representing shapes (i.e. the result of calling tf.shape on another Tensor) this computes a Tensor which is the shape of the result of a broadcasting op applied in tensors of shapes shape_x and shape_y.
For example, if shape_x is [1, 2, 3] and shape_y is [5, 1, 3], the result is a Tensor whose value is [5, 2, 3].
This is useful when validating the result of a broadcasting operation when the tensors do not have statically known shapes.
Args | |
---|---|
shape_x | A rank 1 integer Tensor , representing the shape of x. |
shape_y | A rank 1 integer Tensor , representing the shape of y. |
Returns | |
---|---|
A rank 1 integer Tensor representing the broadcasted shape. |
© 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.3/api_docs/python/tf/broadcast_dynamic_shape