TensorFlow函數(shù)教程：tf.keras.backend.repeat_elements

tf.keras.backend.repeat_elements函數(shù)

tf.keras.backend.repeat_elements(
    x,
    rep,
    axis
)

定義在：tensorflow/python/keras/backend.py。

沿著軸重復張量的元素，比如np.repeat。

如果x具有shape(s1, s2, s3)，并且axis是1，則輸出將具有shape(s1, s2 * rep, s3)。

參數(shù)：

• x：張量或變量。
• rep：Python整數(shù)，要重復的次數(shù)。
• axis：沿著該軸進行重復。

返回：

返回張量。