only one element tensors can be converted to Python scalars

 

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())

你可能感兴趣的:(python,torch)