python中dict转化为numpy张量
时间: 2023-11-30 22:55:31 浏览: 128
Python-在PyTorch中可视化CNN
可以使用numpy的array函数将dict转化为numpy张量。
例如,假设有以下字典:
```
my_dict = {'a': [1, 2, 3], 'b': [4, 5, 6]}
```
可以使用以下代码将其转换为numpy张量:
```
import numpy as np
my_array = np.array(list(my_dict.values()))
```
输出:
```
array([[1, 2, 3],
[4, 5, 6]])
```
其中,list(my_dict.values())将字典的值作为列表返回,然后使用np.array函数将列表转换为numpy张量。
阅读全文