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

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。

返回:

整数张量。

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

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部