[ 组合替代实现 ]torch.is_nonzero

torch.is_nonzero

torch.is_nonzero(input)

用于判断单个元素是否为 0 或者 False ,当 input 的元素个数不为 1 时,抛出 RuntimeError; Paddle 无此 API,需要组合实现。

转写示例

# PyTorch 写法
torch.is_nonzero(x)

# Paddle 写法
x.astype('bool').item()