oneflow.Tensor.bool

Tensor.bool()

Tensor.bool() is equivalent to Tensor.to(oneflow.bool). 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.bool()
>>> input.dtype
oneflow.bool