oneflow.set_default_tensor_type¶
-
oneflow.
set_default_tensor_type
(tensor_type)¶ Sets the default floating point type for those source operators which create Tensor.
The default floating point type is
oneflow.FloatTensor
.- Parameters
tensor_type (type or string) – The floating point tensor type or its name.
For example:
>>> import oneflow >>> oneflow.set_default_tensor_type(oneflow.FloatTensor) >>> x = oneflow.ones(2, 3) >>> x.dtype oneflow.float32 >>> oneflow.set_default_tensor_type("oneflow.DoubleTensor") >>> x = oneflow.ones(2, 3) >>> x.dtype oneflow.float64 >>> oneflow.set_default_tensor_type(oneflow.FloatTensor) >>> x = oneflow.tensor([1.0, 2]) >>> x.dtype oneflow.float32