torch.Tensor.type()方法如何使用

今天就跟大家聊聊有關(guān)torch.Tensor.type()方法如何使用，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結(jié)了以下內(nèi)容，希望大家根據(jù)這篇文章可以有所收獲。

總結(jié):

該方法的功能是:
	當(dāng)不指定dtype時(shí),返回類型.
	當(dāng)指定dtype時(shí),返回類型轉(zhuǎn)換后的數(shù)據(jù),如果類型已經(jīng)符合要求,
	那么不做額外的復(fù)制,返回原對(duì)象.

Microsoft Windows [版本 10.0.18363.1256](c) 2019 Microsoft Corporation。保留所有權(quán)利。

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> a = torch.ones(2,3)>>> a.type()'torch.FloatTensor'>>> a.dtype
torch.float32>>>>>> a
tensor([[1., 1., 1.],[1., 1., 1.]])>>>>>>>>>>>>>>> b = a.type(dtype='torch.DoubleTensor')>>> b
tensor([[1., 1., 1.],[1., 1., 1.]], dtype=torch.float64)>>> b.type()'torch.DoubleTensor'>>> b.dtype
torch.float64>>>>>>>>>>>> b = a.type(dtype='torch.cuda.DoubleTensor')>>> b
tensor([[1., 1., 1.],[1., 1., 1.]], device='cuda:0', dtype=torch.float64)>>> b.type()'torch.cuda.DoubleTensor'>>> b.dtype
torch.float64>>>>>>>>> a
tensor([[1., 1., 1.],[1., 1., 1.]])>>> a.type()'torch.FloatTensor'>>> a.dtype
torch.float32>>>>>>>>>>>>

看完上述內(nèi)容，你們對(duì)torch.Tensor.type()方法如何使用有進(jìn)一步的了解嗎？如果還想了解更多知識(shí)或者相關(guān)內(nèi)容，請(qǐng)關(guān)注億速云行業(yè)資訊頻道，感謝大家的支持。