首頁 > 後端開發 > Python教學 > PyTorch 中的 linspace

PyTorch 中的 linspace

Susan Sarandon
發布: 2025-01-01 13:15:10
原創
535 人瀏覽過

linspace in PyTorch

請我喝杯咖啡☕

*備忘錄:

  • 我的帖子解釋了 arange()。
  • 我的貼文解釋了 logspace()。

linspace() 可以建立零個或多個在 start 和 end 之間均勻間隔的整數、浮點數或複數的一維張量(start

*備忘錄:

  • linspace() 可以與 torch 一起使用,但不能與張量一起使用。
  • torch 的第一個參數是 start(必要型別:int、float、complex 或 bool)。 *int、float、complex 或 bool 的 0D 張量也適用。
  • torch 的第二個參數是 end(必要型別:int、float、complex 或 bool)。 *int、float、complex 或 bool 的 0D 張量也適用。
  • torch的第三個參數是steps(Required-Type:int): *備註:
    • 必須大於或等於0。
    • int 的 0D 張量也適用。
  • torch 有 dtype 參數(可選-預設:無型別:dtype): *備註:
    • 如果為None,則從start、end或step推斷,然後對於浮點數,使用get_default_dtype()。 *我的貼文解釋了 get_default_dtype() 和 set_default_dtype()。
    • 設定整數類型的開始和結束不足以創建整數類型的一維張量,因此必須設定帶有 dtype 的整數類型。
    • 必須使用 dtype=。
    • 我的帖子解釋了 dtype 參數。
  • torch 有裝置參數(可選-預設:無-型別:str、int 或 device()): *備註:
    • 如果為 None,則使用 get_default_device()。 *我的貼文解釋了 get_default_device() 和 set_default_device()。
    • 必須使用 device=。
    • 我的帖子解釋了設備參數。
  • torch 有 require_grad 參數(可選-預設:False-Type:bool): *備註:
    • require_grad=必須使用。
    • 我的帖子解釋了 require_grad 參數。
  • torch 存在 out 參數(可選-預設:無-型別:張量): *備註:
    • 必須使用 out=。
    • 我的貼文解釋了論點。
import torch

torch.linspace(start=10, end=20, steps=0)
torch.linspace(start=20, end=10, steps=0)
# tensor([])

torch.linspace(start=10., end=20., steps=1)
tensor([10.])

torch.linspace(start=20, end=10, steps=1)
# tensor([20.])

torch.linspace(start=10., end=20., steps=2)
# tensor([10., 20.])

torch.linspace(start=20, end=10, steps=2)
# tensor([20., 10.])

torch.linspace(start=10., end=20., steps=3)
# tensor([10., 15., 20.])

torch.linspace(start=20, end=10, steps=3)
# tensor([20., 15., 10.])

torch.linspace(start=10., end=20., steps=4)
# tensor([10.0000, 13.3333, 16.6667, 20.0000])

torch.linspace(start=20., end=10., steps=4)
# tensor([20.0000, 16.6667, 13.3333, 10.0000])

torch.linspace(start=10, end=20, steps=4, dtype=torch.int64)
torch.linspace(start=torch.tensor(10),
               end=torch.tensor(20),
               steps=torch.tensor(4),
               dtype=torch.int64)
# tensor([10.0000, 13.3333, 16.6667, 20.0000])

torch.linspace(start=10.+6.j, end=20.+3.j, steps=4)
torch.linspace(start=torch.tensor(10.+6.j),
               end=torch.tensor(20.+3.j),
               steps=torch.tensor(4))
# tensor([10.0000+6.j, 13.3333+5.j, 16.6667+4.j, 20.0000+3.j])

torch.linspace(start=False, end=True, steps=4)
torch.linspace(start=torch.tensor(True),
               end=torch.tensor(False),
               steps=torch.tensor(4))
# tensor([0.0000, 0.3333, 0.6667, 1.0000])

torch.linspace(start=10, end=20, steps=4, dtype=torch.int64)
torch.linspace(start=torch.tensor(10),
               end=torch.tensor(20),
               steps=torch.tensor(4), dtype=torch.int64)
# tensor([10.0000, 13.3333, 16.6667, 20.0000])
登入後複製

以上是PyTorch 中的 linspace的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:dev.to
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板