tf.range函數(shù):創(chuàng)建數(shù)字序列

2018-11-30 11:00 更新
tf.range 函數(shù)
range(limit, delta=1, dtype=None, name='range')
range(start, limit, delta=1, dtype=None, name='range')

該函數(shù)的使用定義在:tensorflow/python/ops/math_ops.py.

請(qǐng)參閱指南:生成常量,序列和隨機(jī)值>序列

創(chuàng)建一個(gè)數(shù)字序列.

創(chuàng)建一個(gè)數(shù)字序列,該數(shù)字開始于 start 并且將 delta 增量擴(kuò)展到不包括 limit 的序列.

除非明確提供,否則得到的張量的 dtype 是從輸入中推斷出來的.

像 Python 內(nèi)置的 range,start 默認(rèn)為 0,所以 range(n) = range(0, n).

例如:

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]

參數(shù):

  • start:0-D Tensor(標(biāo)量).如果 limit 不是 None,則作為該范圍中的第一個(gè)條目;否則,作為范圍限制,并且第一項(xiàng)默認(rèn)為0.
  • limit:0-D Tensor(標(biāo)量).序列上限,具有排他性.如果沒有,則默認(rèn)值為 start,但是該范圍的第一個(gè)條目默認(rèn)為0.
  • delta:0-D Tensor(標(biāo)量).遞增 start 的數(shù)字.默認(rèn)為1.
  • dtype:結(jié)果張量的元素的類型.
  • name:操作的名稱.默認(rèn)為“range”.

返回:

dtype 類型的一維 Tensor.

numpy 兼容性

相當(dāng)于 np.arange

以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)