torch.is_nonzero(input)
用于判断单个元素是否为 0 或者 False ,当 input 的元素个数不为 1 时,抛出 RuntimeError; Paddle 无此 API,需要组合实现。
0
False
input
RuntimeError
# PyTorch 写法 torch.is_nonzero(x) # Paddle 写法 x.astype('bool').item()