CodeNodeX AI
注册写作
TUTORIALAI 与数据科学FETCHED

PyTorch torch.is_nonzero 函数

PyTorch torch.is_nonzero 函数 Pytorch torch 参考手册 torch.is_nonzero 是 PyTorch 中用于检查张量是否为非零单一元素的函数。 函数定义 torch.is_nonzero(input) 使用示例 实例 [mycode4 type='python'] import torch print(torch.is_nonzero(torch.t..

更新于 2026/7/16
PyTorch

教程详情页默认 noindex。

PyTorch torch.is_nonzero 函数

Pytorch torch 参考手册 Pytorch torch 参考手册

torch.is_nonzero 是 PyTorch 中用于检查张量是否为非零单一元素的函数。

函数定义

torch.is_nonzero(input)

使用示例

实例

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 参考手册