當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。