oneflow.set_default_dtype

oneflow.set_default_dtype()None

Sets the default floating point type for those source operators which create Tensor.

The default floating point type is oneflow.float32.

Parameters

dtype (oneflow.dtype) – The floating point dtype.

For example:

>>> import oneflow
>>> oneflow.set_default_dtype(oneflow.float64)
>>> x = oneflow.randn(2, 3)
>>> x.dtype
oneflow.float64
>>> oneflow.set_default_dtype(oneflow.float32)
>>> x = oneflow.randn(2, 3)
>>> x.dtype
oneflow.float32