當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。