sophia_tianyifei的博客

Torch.tensor.eq() 返回张量维度 描述：分类问题中，常用eq()函数来统计正确样本个数，若张量维度不严格相同时，返回的矩阵维度会发生变化 例：a与b同为1维张量（shape=3），c为2维张量（shape=[3,1]） a = torch.tensor([1,2,3]) b = torch.tensor([1,2,2]) c = torch.tensor([[1],[2],[3]]) 1. 维度相同 a与b维度相同，eq()函数返回维度相同的张量（shape=3），可用于后续计数 >