最近查看tensor输出的时候,只想看到小数后四位,但是由于tensor输出的是float64位的科学技术法,看起来很难受,如下图所示:

1.去掉e显示
由于上图很难看出来区别,所以需要转换一下输出方式,去掉e这种表示,在最上面设置一下
np.set_printoptions(suppress=True)

2.设置四位小数显示
由于小数太多不美观,可以设置只显示4位,代码如下:
np.set_printoptions(precision=4)
效果为:

这样就会美观好多,大家以后可以常用这两项设置