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