当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.range用法及代码示例


创建一个数字序列。

用法

tf.range(limit, delta=1, dtype=None, name='range')
tf.range(start, limit, delta=1, dtype=None, name='range')

参数

  • start 一个 0-D Tensor(标量)。如果 limit 不是 None,则作为范围中的第一个条目;否则,作为范围限制,第一个条目默认为 0。
  • limit 一个 0-D Tensor(标量)。序列的上限,独占。如果为 None,则默认为 start 的值,而范围的第一个条目默认为 0。
  • delta 一个 0-D Tensor(标量)。递增的数字 start 。默认为 1。
  • dtype 结果张量的元素类型。
  • name 操作的名称。默认为"range"。

返回

  • dtype 类型的一维 Tensor

创建从 start 开始并以 delta 为增量扩展的数字序列,直到但不包括 limit

除非明确提供,否则结果张量的 dtype 是从输入中推断出来的。

就像 Python 内置的 range , start 默认为 0,所以 range(n) = range(0, n)

例如:

start = 3
limit = 18
delta = 3
tf.range(start, limit, delta)
<tf.Tensor:shape=(5,), dtype=int32,
numpy=array([ 3,  6,  9, 12, 15], dtype=int32)>
start = 3
limit = 1
delta = -0.5
tf.range(start, limit, delta)
<tf.Tensor:shape=(4,), dtype=float32,
numpy=array([3. , 2.5, 2. , 1.5], dtype=float32)>
limit = 5
tf.range(limit)
<tf.Tensor:shape=(5,), dtype=int32,
numpy=array([0, 1, 2, 3, 4], dtype=int32)>

numpy 兼容性

相当于 np.arange

相关用法


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