pytorch每日一学6(torch.set_default_dtype(d))

本文介绍如何使用PyTorch中的torch.set_default_dtype方法来指定默认浮点数类型,如torch.float32或torch.float64,并通过示例展示了不同情况下浮点数类型的变化。

6. 第六个方法

torch.set_default_dtype(d)
  • 此方法不是很直观,但是比较好理解,意思是设置pytorch中浮点数的默认类型。pytorch中有很多浮点类型,例如torch.float16、torch.float32、torch.float64这些在初始化一个浮点tensor的时候是可以指定的,如果我们不指定,那么pytorch就是默认给其一个类型,此方法的作用就是指定pytorch默认给不指定浮点类型的浮点数哪个类型。我们先来看个例子。
import torch
a = torch.tensor([1., 2.])
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值