pytorch(9)-item()

使用积累 item() item() 一个Tensor调用.item()方法就可以返回这个Tensor 对应的标准python 类型的数据.注意事项,只针对一个元素的标量.若是向量,可以调用tolist()方法.
本站公众号
   欢迎关注本站公众号,获取更多信息