only one element tensors can be converted to Python scalars
for neg_step in range(neg_,float(con_pos>=0.6).sum()//neg_+1,neg_):
ValueError: only one element tensors can be converted to Python scalars
出错原因:
tensor不能直接转成float类型
正确写法:
sum()能转类型,应该转成int型。
float((con_pos>=0.6).sum())