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.
Тип результата (dtype) выводится из входных данных, если не указан явно.
Как и встроенная функция 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". |
| Возвращаемые значения | |
|---|---|
1-мерный Tensor типа dtype. |
Совместимость с NumPy
Эквивалентно np.arange
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/range