python将numpy64位转化为tensor
时间: 2023-08-15 08:06:03 浏览: 138
Pytorch之Tensor和Numpy之间的转换的实现方法
要将numpy数组转换为Tensor,你可以使用PyTorch库中的`torch.from_numpy()`函数。下面是一个示例代码:
```python
import numpy as np
import torch
# 创建一个numpy数组
numpy_array = np.array([1, 2, 3, 4, 5], dtype=np.float64)
# 将numpy数组转换为tensor
tensor = torch.from_numpy(numpy_array)
print(tensor)
```
这将把一个64位浮点型的numpy数组转换为PyTorch中的浮点型Tensor。你可以通过打印`tensor`来查看结果。
阅读全文