torch.sum(tensor[index]==0).item()

物物不物于物

已于 2023-02-08 16:19:09 修改

阅读量188

点赞数

文章标签：人工智能深度学习 python Powered by 金山文档

于 2023-02-08 10:27:23 首次发布

本文链接：https://blog.csdn.net/qq_50799276/article/details/128930979

版权

a

# tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])

sum = torch.sum(a[1]==0)
sum

# tensor(0)

sum = torch.sum(a[1]==0).item()
sum

# 0

a[0]==0

# tensor([ True, False, False, False])

sum = torch.sum(a[0]==0)
sum

# tensor(1)

sum = torch.sum(a[0]==0).item()
sum

# 1

a1 = torch.zeros(3,4)
a1

# tensor([[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [0., 0., 0., 0.]])

sum1 = torch.sum(a1[0]==0.0)
sum1

# tensor(4)

sum1 = torch.sum(a1[0]==0.0).item()
sum1

# 4