oneflow.Tensor.float

Tensor.float()

Tensor.float() is equivalent to Tensor.to(flow.float32). See oneflow.Tensor.to().

Parameters

input (Tensor) – the input tensor.

For example:

>>> import oneflow as flow
>>> import numpy as np

>>> input = flow.tensor(np.random.randn(1, 2, 3), dtype=flow.int)
>>> input = input.float()
>>> input.dtype
oneflow.float32