pytorch如何转换tensor的类型dtype
比如现在要将 float 类型的 tensor 转换为 long 类型：# 两种方法都可以a = a.long()a = torch.tensor(a, dtype=torch.long)同理，如果是改为其他类型，则把 long 替换为对应的类型就行
·
比如现在要将 float 类型的 tensor 转换为 long 类型：
# 两种方法都可以
a = a.long()
a = torch.tensor(a, dtype=torch.long)同理，如果是改为其他类型，则把 long 替换为对应的类型就行
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
9
0- 0
已为社区贡献1条内容
所有评论(0)