ValueError: only one element tensors can be converted to Python scalars
pytorch ,我用的是GPU版
出现了tensor 转 scalars的问题,
如果你的电脑跑的是CPU tensor
执行下面这句 tensor 转numpy;这是cpu版
ndarray = tensor.numpy()
如果你的是GPU版,tensor 不能直接转numpy,还需要先转cpu才行,如下
ndarray = tensor.cpu().numpy()