pytorch中的item()用法

将零维度张量转为浮点数

posted @ 2022-03-20 12:10  小白白中白  阅读(48)  评论(0)  编辑  收藏  举报