温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

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

发布时间:2021-08-05 17:47:23 来源:亿速云 阅读:180 作者:Leah 栏目:大数据

今天就跟大家聊聊有关torch.Tensor.type()方法如何使用,可能很多人都不太了解,为了让大家更加了解,小编给大家总结了以下内容,希望大家根据这篇文章可以有所收获。

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

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

该方法的功能是:
	当不指定dtype时,返回类型.
	当指定dtype时,返回类型转换后的数据,如果类型已经符合要求,
	那么不做额外的复制,返回原对象.
Microsoft Windows [版本 10.0.18363.1256](c) 2019 Microsoft Corporation。保留所有权利。

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>>>>>>>>>>>>

看完上述内容,你们对torch.Tensor.type()方法如何使用有进一步的了解吗?如果还想了解更多知识或者相关内容,请关注亿速云行业资讯频道,感谢大家的支持。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI