主要是将tensor数据类型转为C++可以识别的类型。
拿出tensor.item()之后,还可以继续转,比如C++中不能用if判断torch tensor的类型,但通过tensor.item().toBool()就转为了可以识别的类型了。
同理,也可以to其他类型。
本文介绍如何将PyTorch中的张量(tensor)类型转换为C++可以识别的形式,如使用tensor.item().toBool()实现类型转换,并展示了常见类型转换的方法和应用场景。
主要是将tensor数据类型转为C++可以识别的类型。
拿出tensor.item()之后,还可以继续转,比如C++中不能用if判断torch tensor的类型,但通过tensor.item().toBool()就转为了可以识别的类型了。
同理,也可以to其他类型。

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