View source on GitHub |
Generates evenly-spaced values in an interval along a given axis.
tf.linspace( start, stop, num, name=None, axis=0 )
A sequence of num
evenly-spaced values are generated beginning at start
along 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.
Matches np.linspace's behaviour except when num == 0
.
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)>
Args | |
---|---|
start | A Tensor . Must be one of the following types: bfloat16 , float32 , float64 . N-D tensor. First entry in the range. |
stop | A Tensor . Must have the same type and shape as start . N-D tensor. Last entry in the range. |
num | A Tensor . Must be one of the following types: int32 , int64 . 0-D tensor. Number of values to generate. |
name | A name for the operation (optional). |
axis | Axis along which the operation is performed (used only when N-D tensors are provided). |
Returns | |
---|---|
A Tensor . Has the same type as start . |
© 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.3/api_docs/python/tf/linspace