一、转换时间对比
- 构建100W*100的二维数组,各类数据转tensor时间代价
时间 | 总时间 | |
---|---|---|
ndarray<ndarray> 到 tensor | 0.2138 | 0.2138 |
list<list> 到 tensor | 1.8477 | 1.8477 |
list<list> 到 ndarray<ndarray> 到 tensor | 8.6217 -> 0.1926 | 8.8143 |
list<ndarray> 到 tensor | 11.9110 | 11.911 |
list<ndarray> 到 ndarray<ndarray> 到 tensor | 0.8266 -> 0.1390 | 0.9656 |
- 将list<list>转化为ndarray<ndarray>
时间 | 总时间 | |
---|---|---|
list<list> 到 ndarray<ndarray> | 8.6217 | 8.6217 |
list<list> 到 tensor 到 ndarray<ndarray> | 1.7874 -> 0.1337 | 1.9211 |
- 结论(原因有待分析)
- 尽量使用ndarray转tensor
- list有ndarray,选择list转ndarray转tensor
- list没有ndarray,直接list转tensor