Функция range

Функция range создает одномерный тензор с последовательностью чисел. Первым параметром функция принимает начало диапазона, вторым - конец диапазона (не включая его), третьим - шаг. По умолчанию начало равно нулю, а шаг - единице. Тип данных элементов можно задать параметром dtype.

Синтаксис

tf.range(start, limit, delta, [dtype])

Пример

Давайте создадим тензор с числами от 0 до 5:

import tensorflow as tf

t = tf.range(5)
print(t)

Результат выполнения кода:

tf.Tensor([0 1 2 3 4], shape=(5,), dtype=int32)

Пример

Давайте создадим тензор с числами от 1 до 10 с шагом 2:

import tensorflow as tf

t = tf.range(1, 10, 2)
print(t)

Результат выполнения кода:

tf.Tensor([1 3 5 7 9], shape=(5,), dtype=int32)

Пример

Давайте создадим тензор с типом float32:

import tensorflow as tf

t = tf.range(0, 1, 0.2, dtype=tf.float32)
print(t)

Результат выполнения кода:

tf.Tensor([0. 0.2 0.4 0.6 0.8], shape=(5,), dtype=float32)

Смотрите также

функцию linspace,
которая создает тензор с равномерно распределенными числами
функцию constant,
которая создает тензор из переданных данных
функцию convert_to_tensor,
которая преобразует данные в тензор
функцию zeros,
которая создает тензор, заполненный нулями