torch.item()按照官方文档,可以理解为从只包含一个元素的tensor中提取值,注意是只含有one element,其他情况用tolist()在训练时统计loss的变化时,经常用到, ...
確定! 回上一頁