比如现在要将 float 类型的 tensor 转换为 long 类型：
# 两种方法都可以
a = a.long()
a = torch.tensor(a, dtype=torch.long)
同理，如果是改为其他类型，则把 long 替换为对应的类型就行
pytorch如何转换tensor的类型dtype
阅读 68
2023-02-23
相关推荐
精彩评论（0）
0 0 举报
pytorch如何转换tensor的类型dtype
阅读 68
2023-02-23
比如现在要将 float 类型的 tensor 转换为 long 类型：
# 两种方法都可以
a = a.long()
a = torch.tensor(a, dtype=torch.long)
同理，如果是改为其他类型，则把 long 替换为对应的类型就行
相关推荐
精彩评论（0）