python中将张量转化为整数数组
时间: 2023-10-09 17:06:12 浏览: 152
你可以使用PyTorch中的`.tolist()`函数将张量转换为整数数组。例如:
```python
import torch
# 创建张量
tensor = torch.tensor([1, 2, 3, 4])
# 将张量转化为整数数组
array = tensor.tolist()
print(array) # 输出 [1, 2, 3, 4]
```
你也可以使用`.numpy()`函数将张量转换为NumPy数组,然后使用`.astype()`函数将其转换为整数数组。例如:
```python
import torch
# 创建张量
tensor = torch.tensor([1.2, 2.3, 3.4, 4.5])
# 将张量转化为整数数组
array = tensor.numpy().astype(int)
print(array) # 输出 [1, 2, 3, 4]
```
阅读全文