Python PyTorch range用法及代碼示例

本文簡要介紹python語言中 torch.range 的用法。

用法:

torch.range(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

參數：

• start(float) -點集的起始值。默認值：0。

• end(float) -點集的結束值

• step(float) -每對相鄰點之間的間隙。默認值：1。

關鍵字參數：

• out(Tensor,可選的) -輸出張量。

• dtype(torch.dtype， 可選的) -返回張量的所需數據類型。默認值：如果 None ，使用全局默認值(參見 torch.set_default_tensor_type() )。如果未給出 dtype，則從其他輸入參數推斷數據類型。如果 start 、 end 或 stop 中的任何一個是浮點數，則 dtype 被推斷為默認 dtype，請參閱 get_default_dtype() 。否則， dtype 被推斷為 torch.int64 。

• layout(torch.layout， 可選的) -返回張量的所需布局。默認值：torch.strided。

• device(torch.device， 可選的) -返回張量的所需設備。默認值：如果 None ，使用當前設備作為默認張量類型(參見 torch.set_default_tensor_type() )。 device 將是 CPU 張量類型的 CPU 和 CUDA 張量類型的當前 CUDA 設備。

• requires_grad(bool,可選的) -如果 autograd 應該在返回的張量上記錄操作。默認值：False。

返回大小為 \left\lfloor \frac{\text{end} - \text{start}}{\text{step}} \right\rfloor + 1 的一維張量，其值從 start 到 end，步驟為 step 。步長是張量中兩個值之間的差距。

\text{out}_{i+1} = \text{out}_i + \text{step}.

警告

此函數已棄用，並將在未來的版本中刪除，因為它的行為與 Python 的內置範圍不一致。相反，請使用 torch.arange() ，它會在 [start, end) 中生成值。

例子：

>>> torch.range(1, 4)
tensor([ 1.,  2.,  3.,  4.])
>>> torch.range(1, 4, 0.5)
tensor([ 1.0000,  1.5000,  2.0000,  2.5000,  3.0000,  3.5000,  4.0000])