解决方案:
可以先输出该变量的类型查看:print(type(a))
如果本来就是numpy类型,直接在代码前添加:
import numpy as npnp.set_printoptions(threshold=np.inf) # 全部输出
如果是tensor张量型,就要先把tensor转换为numpy,变量是output:
(output.numpy())
如果报错,可能是因为tensor是GPU上的,要先转换为cpu。
(output.cpu()).numpy()
时间:2024-05-06 23:08:18
可以先输出该变量的类型查看:print(type(a))
如果本来就是numpy类型,直接在代码前添加:
import numpy as npnp.set_printoptions(threshold=np.inf) # 全部输出
如果是tensor张量型,就要先把tensor转换为numpy,变量是output:
(output.numpy())
如果报错,可能是因为tensor是GPU上的,要先转换为cpu。
(output.cpu()).numpy()