使用pytorch张量格式统计一个向量中非0值的个数

可以使用 PyTorch 的 `count_nonzero()` 函数来实现统计一个向量中非零值的个数。以下是一个示例代码：

import torch

# 创建一个向量
x = torch.tensor([1, 0, 3, 0, 5, 0])

# 统计非0值的个数
count = torch.count_nonzero(x)

print(count)

输出结果为：

tensor(3)

说明该向量中有3个非0值。