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

由 Carrie 创建, 最后一次修改 2019-03-05

tf.keras.backend.arange函数

tf.keras.backend.arange(
    start,
    stop=None,
    step=1,
    dtype='int32'
)

定义在:tensorflow/python/keras/backend.py。

创建包含整数序列的1D张量。

函数参数使用与Theano的arange相同的约定:如果只提供一个参数,它实际上是“stop”参数,并且“start”参数为0。

返回张量的默认类型是'int32',匹配TensorFlow的默认值。

参数:

  • start:开始值。
  • stop:停止值。
  • step:两个连续值之间的差异。
  • dtype:要使用的整数dtype。

返回:

整数张量。

以上内容是否对您有帮助:
二维码
建议反馈
二维码