numpy.uint16转tensor
时间: 2023-10-19 13:47:13 浏览: 249
pytorch 实现tensor与numpy数组转换
要将numpy数组的数据类型从`numpy.uint16`转换为PyTorch的Tensor,可以使用`torch.from_numpy()`函数。以下是一个示例:
```python
import numpy as np
import torch
# 创建一个numpy数组
arr = np.array([1, 2, 3], dtype=np.uint16)
# 将numpy数组转换为torch tensor
tensor = torch.from_numpy(arr)
```
在上面的示例中,我们首先创建了一个名为`arr`的numpy数组,数据类型为`numpy.uint16`。然后,我们使用`torch.from_numpy()`函数将该数组转换为PyTorch的Tensor,存储在变量`tensor`中。
请注意,转换后的Tensor将具有与原始numpy数组相同的数据类型。
阅读全文