tf.range
| Просмотреть исходный код на GitHub |
Создаёт последовательность чисел.
tf.range(limit, delta=1, dtype=None, name='range') tf.range(start, limit, delta=1, dtype=None, name='range')
Создаёт последовательность чисел, которая начинается с start и увеличивается на delta до, но не включая limit.
Тип данных результирующего тензора определяется по входным данным, если явно не задан.
Подобно встроенной функции Python range, по умолчанию start равно 0, поэтому range(n) = range(0, n).
Например:
start = 3 limit = 18 delta = 3 tf.range(start, limit, delta) <tf.Tensor: shape=(5,), dtype=int32, numpy=array([ 3, 6, 9, 12, 15], dtype=int32)>
start = 3 limit = 1 delta = -0.5 tf.range(start, limit, delta) <tf.Tensor: shape=(4,), dtype=float32, numpy=array([3. , 2.5, 2. , 1.5], dtype=float32)>
limit = 5 tf.range(limit) <tf.Tensor: shape=(5,), dtype=int32, numpy=array([0, 1, 2, 3, 4], dtype=int32)>
| Аргументы | |
|---|---|
start | 0-мерный Tensor (скаляр). Выступает в качестве первого элемента в последовательности, если limit не равно None; в противном случае, выступает как предел последовательности, а первый элемент по умолчанию равен 0. |
limit | 0-мерный Tensor (скаляр). Верхняя граница последовательности, не включающая это значение. Если None, по умолчанию принимает значение start, при этом первый элемент последовательности по умолчанию равен 0. |
delta | 0-мерный Tensor (скаляр). Значение, на которое увеличивается start. По умолчанию равно 1. |
dtype | Тип элементов результирующего тензора. |
name | Имя операции. По умолчанию равно "range". |
| Возвращаемое значение | |
|---|---|
Одномерный Tensor типа dtype. |
Совместимость с numpy
Эквивалентно np.arange
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/range