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

Python tf.repeat用法及代碼示例


重複 input 的元素。

用法

tf.repeat(
    input, repeats, axis=None, name=None
)

參數

  • input N 維張量。
  • repeats 一維int張量。每個元素的重複次數。廣播重複以適應給定軸的形狀。 len(repeats) 必須等於 input.shape[axis] 如果軸不是無。
  • axis An int. 沿其重複值的軸。默認情況下(axis=None)，使用扁平化輸入數組，並返回扁平化輸出數組。
  • name 操作的名稱。

返回

  • 與 input 具有相同形狀的張量，除了沿給定軸。如果axis為None，則輸出數組被展平以匹配展平的輸入數組。

另見tf.concat、tf.stack、tf.tile。

示例用法：

repeat(['a', 'b', 'c'], repeats=[3, 0, 2], axis=0)
<tf.Tensor:shape=(5,), dtype=string,
numpy=array([b'a', b'a', b'a', b'c', b'c'], dtype=object)>
repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=0)
<tf.Tensor:shape=(5, 2), dtype=int32, numpy=
array([[1, 2],
       [1, 2],
       [3, 4],
       [3, 4],
       [3, 4]], dtype=int32)>
repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=1)
<tf.Tensor:shape=(2, 5), dtype=int32, numpy=
array([[1, 1, 2, 2, 2],
       [3, 3, 4, 4, 4]], dtype=int32)>
repeat(3, repeats=4)
<tf.Tensor:shape=(4,), dtype=int32, numpy=array([3, 3, 3, 3], dtype=int32)>
repeat([[1,2], [3,4]], repeats=2)
<tf.Tensor:shape=(8,), dtype=int32,
numpy=array([1, 1, 2, 2, 3, 3, 4, 4], dtype=int32)>

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.repeat。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。