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

物物不物于物

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

阅读量260

点赞数

文章标签：人工智能深度学习 python

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

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

版权

文章展示了如何在PyTorch中使用tensor进行条件求和操作，以及将结果转换为标量值。通过示例代码，解释了torch.sum函数的用法，特别是在比较操作后进行求和的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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