tf.linspace  |  TensorFlow v2.16.1 (original) (raw)

tf.linspace

Stay organized with collections Save and categorize content based on your preferences.

Generates evenly-spaced values in an interval along a given axis.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.lin_space, tf.compat.v1.linspace

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

Used in the notebooks

A sequence of num evenly-spaced values are generated beginning at startalong a given axis. If num > 1, the values in the sequence increase by(stop - start) / (num - 1), so that the last one is exactly stop. If num <= 0, ValueError is raised.

Matchesnp.linspace's behaviour except when num == 0.

For example:

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

Start and stop can be tensors of arbitrary size:

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 is where the values will be generated (the dimension in the returned tensor which corresponds to the axis will be equal to 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)>