torch.cuda.ByteTensor转为torch.cuda.FloatTensor
时间: 2023-12-04 12:57:17 浏览: 166
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
可以通过以下代码将torch.cuda.ByteTensor转换为torch.cuda.FloatTensor:
```
import torch
# 创建ByteTensor
x = torch.cuda.ByteTensor(2, 3).random_(0, 2)
# 转换为FloatTensor
y = x.float()
print(x)
print(y)
```
输出结果:
```
tensor([[0, 0, 1],
[0, 1, 1]], device='cuda:0', dtype=torch.uint8)
tensor([[0., 0., 1.],
[0., 1., 1.]], device='cuda:0')
```
阅读全文