tf.linspace

Генерирует равномерно распределённые значения в интервале вдоль заданной оси.

tf.linspace(
    start, stop, num, name=None, axis=0
)

Используется в блокнотах

Генерируется последовательность из num равномерно распределённых значений, начиная с start вдоль заданной axis. Если num > 1, значения в последовательности увеличиваются на (stop - start) / (num - 1), так что последнее значение точно равно stop. Если num <= 0, то ValueError.

Соответствует поведению np.linspace, за исключением случаев, когда num == 0.

Например:

tf.linspace(10.0, 12.0, 3, name="linspace") => [ 10.0  11.0  12.0]

Start и stop могут быть тензорами произвольного размера:

tf.linspace([0., 5.], [10., 40.], 5, axis=0)
<tf.Tensor: shape=(5, 2), dtype=float32, numpy=
array([[ 0.  ,  5.  ],
       [ 2.5 , 13.75],
       [ 5.  , 22.5 ],
       [ 7.5 , 31.25],
       [10.  , 40.  ]], dtype=float32)>

Axis — это место, где будут сгенерированы значения (размерность возвращаемого тензора, соответствующая оси, будет равна num).

tf.linspace([0., 5.], [10., 40.], 5, axis=-1)
<tf.Tensor: shape=(2, 5), dtype=float32, numpy=
array([[ 0.  ,  2.5 ,  5.  ,  7.5 , 10.  ],
       [ 5.  , 13.75, 22.5 , 31.25, 40.  ]], dtype=float32)>