首页 > 其他 > 详细

Pytorch

时间:2021-04-16 12:44:38      阅读:22      评论:0      收藏:0      [点我收藏+]

pytorch 模型的断点继训

pytorch .item()作用

Use torch.Tensor.item() to get a Python number from a tensor containing a single value.

.item()方法返回张量元素的值。

>>> import torch
>>> x = torch.tensor([[1]])
>>> x
tensor([[1]])
>>> x.item()
1
>>> x = torch.tensor(2.5)
>>> x
tensor(2.5000)
>>> x.item()
2.5

 

Pytorch

原文:https://www.cnblogs.com/Overture/p/14665433.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!