0
点赞
收藏
分享

微信扫一扫

基于SSM的理发店会员管理系统的设计和实现(有报告)。Javaee项目。ssm项目。

本文重点

前面我们学习了pytorch的默认数据类型是torch.FloatTensor,那么本节课程我们将学习如何将不同的数据类型之间进行数据转换?还会学习tensor中cpu和gpu的转换,以及tensor和numpy以及python之间的类型转换,在实际编程中,可能这些需要时刻使用的。

不同类型之间的转换

不同类型之间的转换非常简单,只需要在Tensor后加 .long(), .int(), .float(), .double()等即可,也可以用.to()函数进行转换

tensor = torch.Tensor(3, 5)

CPU和GPU之间的转换

举报

相关推荐

0 条评论