pytorch如何统计张量中小于0的数量

可以使用以下代码来统计 PyTorch 张量中小于 0 的元素数量：

import torch

x = torch.randn(3, 4)
num_negatives = (x < 0).sum().item()
print("Number of negative elements:", num_negatives)

解释一下代码：

首先，我们创建一个大小为 3x4 的张量 `x`，里面的元素是从标准正态分布中随机采样得到的。

然后，我们使用 `(x < 0)` 来创建一个大小与 `x` 相同的布尔类型的张量，表示 `x` 中每个元素是否小于 0。

接着，我们使用 `.sum()` 方法对布尔张量中的元素求和，得到小于 0 的元素数量。

最后，我们使用 `.item()` 方法将数量转换为 Python 标量并打印出来。