torch中astype函数的用法
时间: 2023-05-16 10:06:26 浏览: 335
torch 中各种图像格式转换的实现方法
astype函数用于将张量的数据类型转换为指定的数据类型。它的语法如下:
```python
torch.Tensor.astype(dtype)
```
其中,dtype是指定的数据类型,可以是torch.float、torch.double、torch.int等。
例如,将一个张量转换为float类型:
```python
import torch
x = torch.tensor([1, 2, 3])
x = x.astype(torch.float)
print(x)
```
输出:
```
tensor([1., 2., 3.])
```
阅读全文