在使用torch 包的时候,需要找到数组中为0的数量
b_ = [i for i in a_ if i == 0]
如果 a_ 是 torch.tensor 类型的,那慢死你
如果 a_ 是 numpy.array 类型的,还好,至少不会卡死
在使用torch 包的时候,需要找到数组中为0的数量
b_ = [i for i in a_ if i == 0]
如果 a_ 是 torch.tensor 类型的,那慢死你
如果 a_ 是 numpy.array 类型的,还好,至少不会卡死