TensorFlow函数教程:tf.keras.backend.repeat_elements

tf.keras.backend.repeat_elements函数

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)。

参数:

  • x:张量或变量。
  • rep:Python整数,要重复的次数。
  • axis:沿着该轴进行重复。

返回:

返回张量。

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部