PyTorch torch.is_nonzero Function

Pytorch torch 参考手册Pytorch torch Reference Manual

torch.is_nonzeroIt is a function in PyTorch used to check whether a tensor is a non-zero single element.

Function Definition

torch.is_nonzero(input)

Usage Example

Example

import torch

print(torch.is_nonzero(torch.tensor([1.0])))
print(torch.is_nonzero(torch.tensor([0.0])))
print(torch.is_nonzero(torch.tensor([1, 2])))

Pytorch torch 参考手册Pytorch torch Reference Manual

Other Extensions