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

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

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

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

創(chuàng)建包含整數(shù)序列的1D張量。

函數(shù)參數(shù)使用與Theano的arange相同的約定：如果只提供一個參數(shù)，它實際上是“stop”參數(shù)，并且“start”參數(shù)為0。

返回張量的默認類型是'int32'，匹配TensorFlow的默認值。

參數(shù)：

• start：開始值。
• stop：停止值。
• step：兩個連續(xù)值之間的差異。
• dtype：要使用的整數(shù)dtype。

返回：

整數(shù)張量。