我有一个带形状的矩阵(1255,13,13),所以我有13x13=169个元素,对于每个元素,我有一个由255个元素组成的数组。
我想迭代255数组的第四个元素,并计算有多少元素大于0.5。
此代码不起作用,但有助于理解我想要的内容:
out = net.forward()
count1=0
count2=0
for i in out[0]:
for j in out[2]:
for a in out[3]:
for b in out[1]:
if b[3]> 0.5:
count1+=1
else:
count2+=1
最好的方法是什么?