在学习沐神的《动手学深度学习》中,遇到如下语句
return rmse.item()
不知何用,经过检索发现.item()用于在只包含一个元素的tensor中提取值,并且返回python数据类型
,注意是只包含一个元素,否则的话使用.tolist()
在学习沐神的《动手学深度学习》中,遇到如下语句
return rmse.item()
不知何用,经过检索发现.item()用于在只包含一个元素的tensor中提取值,并且返回python数据类型
,注意是只包含一个元素,否则的话使用.tolist()
8万+
1987

被折叠的 条评论
为什么被折叠?
>