方法:
torch.set_default_dtype(d)
功能: 设置当前环境下浮点数的默认类型
torch.get_default_dtype(d) 获取当前默认浮点类型
注意:
1. 这个设置不是永久的,只在当前环境下有用
2. 这个设置只对没有设置浮点类型的元素有用
例子:
方法:
torch.set_default_dtype(d)
功能: 设置当前环境下浮点数的默认类型
torch.get_default_dtype(d) 获取当前默认浮点类型
注意:
1. 这个设置不是永久的,只在当前环境下有用
2. 这个设置只对没有设置浮点类型的元素有用
例子: