any() missing 1 required positional arguments: "dim"

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/82801149

any() missing 1 required positional arguments: "dim"

scores = torch.linspace(0.5, 0.6, steps=2).view(2)
# b = torch.FloatTensor(0.5).seqence()
# print(b)

if np.any(scores != -1):

   print(111)

解决方法:

np.any(np.array(scores >5))

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/82801149