当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.broadcast_dynamic_shape用法及代码示例


计算给定符号形状的广播的形状。

用法

tf.broadcast_dynamic_shape(
    shape_x, shape_y
)

参数

  • shape_x 秩 1 整数 Tensor ，表示 x 的形状。
  • shape_y 秩 1 整数 Tensor ，表示 y 的形状。

返回

  • 代表广播形状的等级 1 整数Tensor。

抛出

  • InvalidArgumentError 如果这两种形状不适合广播。

当 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], ...>

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.broadcast_dynamic_shape。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。