tf.range()

2022-10-06 08:27:44 浏览数 (1)

参考tf.range - 云 社区 - 腾讯云

代码语言:javascript复制
tf.range(limit, delta=1, dtype=None, name='range')
tf.range(start, limit, delta=1, dtype=None, name='range')

创建一个数字序列。创建一个从start开始的数字序列,并以增量形式扩展,直到但不包括limit。得到的张量的d型是由输入推断出来的,除非它是显式提供的。与Python内置范围一样,start默认值为0,因此该range(n) =range(0,n)。 例如:

代码语言:javascript复制
start = 3
limit = 18
delta = 3
tf.range(start, limit, delta)  # [3, 6, 9, 12, 15]

start = 3
limit = 1
delta = -0.5
tf.range(start, limit, delta)  # [3, 2.5, 2, 1.5]

limit = 5
tf.range(limit)  # [0, 1, 2, 3, 4]

参数:

  • start:   一个0-D张量(标量)。如果limit不为零,则作为范围内的第一个项;否则,作为范围限制,第一个条目默认为0。
  • limit:   一个0-D张量(标量)。序列的上限,排他。如果没有,则默认值为start,而范围的第一个条目默认值为0。
  • delta:   一个0-D张量(标量)。开始递增的数字。默认为1。
  • dtype:   得到的张量的元素的类型。
  • name:     操作的名称。默认为“range”。

返回值:

  • dtype类型的一维张量。

0 人点赞