pytorch.range()和pytorch.arange()的區(qū)別及說明
pytorch.range()和pytorch.arange()的區(qū)別
>>> y=torch.range(1,6) >>> y tensor([1., 2., 3., 4., 5., 6.]) >>> y.dtype torch.float32 >>> z=torch.arange(1,6) >>> z tensor([1, 2, 3, 4, 5]) >>> z.dtype torch.int64
心得
torch.range(start=1, end=6)
的結(jié)果是會包含end
的,而torch.arange(start=1, end=6)
的結(jié)果并不包含end
。- 兩者創(chuàng)建的
tensor
的類型也不一樣。
參考:
Pytorch torch.range()&torch.arange()
torch.range(start=1, end=6)
結(jié)果是會包含end的,創(chuàng)建的tensor的類型為float32。
torch.arange(start=1, end=6)
結(jié)果并不包含end,創(chuàng)建的tensor的類型為int64。
代碼
>>> y=torch.range(1,6) >>> y tensor([1., 2., 3., 4., 5., 6.]) >>> y.dtype torch.float32 >>> z=torch.arange(1,6) >>> z tensor([1, 2, 3, 4, 5]) >>> z.dtype torch.int64
pytorch torch.arange
應(yīng)用
>>> torch.arange(5) tensor([ 0, 1, 2, 3, 4])
API
torch.arange(start=0, end, step=1, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
參數(shù) | 描述 |
---|---|
start (Number) | |
end (Number) | |
step (Number) | |
out (Tensor, optional) | |
dtype (torch.dtype, optional) | |
layout (torch.layout, optional) | |
device (torch.device, optional) | |
requires_grad (bool, optional) |
總結(jié)
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
- PyTorch中torch.load()的用法和應(yīng)用
- python中torch.load中的map_location參數(shù)使用
- Pytorch中的torch.nn.Linear()方法用法解讀
- Pytorch中的torch.where函數(shù)使用
- python中的List sort()與torch.sort()
- 關(guān)于torch.scatter與torch_scatter庫的使用整理
- PyTorch函數(shù)torch.cat與torch.stac的區(qū)別小結(jié)
- 使用with torch.no_grad():顯著減少測試時顯存占用
- PyTorch中torch.save()的用法和應(yīng)用小結(jié)
相關(guān)文章
tensorflow基于CNN實(shí)戰(zhàn)mnist手寫識別(小白必看)
這篇文章主要介紹了tensorflow基于CNN實(shí)戰(zhàn)mnist手寫識別(小白必看),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07Python遠(yuǎn)程開發(fā)環(huán)境部署與調(diào)試過程圖解
這篇文章主要介紹了Python遠(yuǎn)程開發(fā)環(huán)境部署與調(diào)試過程圖解,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-12-12pygame實(shí)現(xiàn)俄羅斯方塊游戲(對戰(zhàn)篇1)
這篇文章主要為大家詳細(xì)介紹了pygame實(shí)現(xiàn)俄羅斯方塊游戲的對戰(zhàn)篇,文中示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下2019-10-10pymssql數(shù)據(jù)庫操作MSSQL2005實(shí)例分析
這篇文章主要介紹了pymssql數(shù)據(jù)庫操作MSSQL2005的方法,可實(shí)現(xiàn)基本的連接、查詢、插入、更新及調(diào)用存儲過程等功能,非常具有實(shí)用價值,需要的朋友可以參考下2015-05-05python selenium禁止加載某些請求的實(shí)現(xiàn)
本文主要介紹了python selenium禁止加載某些請求的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下2022-01-01python+html文字點(diǎn)選驗證碼加固安全防線
這篇文章主要為大家介紹了python文字點(diǎn)選驗證碼加固安全防線實(shí)現(xiàn)示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-09-09