Python tf.linspace用法及代码示例

在给定轴的间隔中生成 evenly-spaced 值。

用法

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

参数

start 一个Tensor。必须是以下类型之一：bfloat16 , float32 , float64。 N-D 张量。范围内的第一个条目。
stop 一个Tensor。必须具有与 start 相同的类型和形状。 N-D 张量。范围内的最后一个条目。
num 一个Tensor。必须是以下类型之一：int32 , int64。 0-D 张量。要生成的值的数量。
name 操作的名称(可选)。
axis 执行操作的轴(仅在提供N-D 张量时使用)。

一个Tensor。具有与 start 相同的类型。

沿着给定的 axis 从 start 开始生成一系列 num evenly-spaced 值。如果 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)>

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.linspace。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。