oneflow.Tensor.long

Tensor.long()

Tensor.long() is equivalent to Tensor.to(flow.int64). 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.float32)
>>> input = input.long()
>>> input.dtype
oneflow.int64