TensorFlow函数教程:tf.keras.backend.arange
2019-03-06 16:09 更新
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。
返回:
整数张量。