tensor&ndarray&int、float

柔情只为你懂 2023-06-04 12:51 40阅读 0赞

(1)如果tensor只有一个元素,然后转换成int或者float类型的时候直接用int()或者float()就可以了;

(2)如果tensor含有多个元素,转换成ndarray时就要用x.detach().numpy()操作了(视情况使用,cpu());

转载于:https://www.cnblogs.com/zf-blog/p/11365083.html

发表评论

表情:
评论列表 (有 0 条评论,40人围观)

还没有评论,来说两句吧...

相关阅读