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