| 面向TensorFlow和PyTorch的线性代数基准测试 |
| |
| 引用本文： | 胥凌. 面向TensorFlow和PyTorch的线性代数基准测试[J]. 航空计算技术, 2022, 52(3): 5-9. DOI: 10.3969/j.issn.1671-654X.2022.03.002 |
| |
| 作者姓名： | 胥凌 |
| |
| 作者单位： | 航空工业西安航空计算技术研究所,陕西 西安710000 |
| |
| 摘 要： | 传统的高性能线性代数计算库如BLAS需要开发者具备丰富的性能优化经验，使用困难。TensorFlow、Pytorch等AI框架提供了简单的开发接口，促进了机器学习应用的发展。这些AI框架大量进行线性代数计算，但是不清楚其是否针对线性代数计算进行了性能优化。设计了一组线性代数计算测试程序，评估了AI框架对的线性代数计算的...
|
| 关 键 词： | 线性代数 TensorFlow Pytorch 高性能计算 矩阵乘法 |
Linear Algebra Benchmarks for TensorFlow and PyTorch |
| |
| XU Ling. Linear Algebra Benchmarks for TensorFlow and PyTorch[J]. Aeronautical Computer Technique, 2022, 52(3): 5-9. DOI: 10.3969/j.issn.1671-654X.2022.03.002 |
| |
| Authors: | XU Ling |
| |
| Abstract: | |
| |
| Keywords: | |
| 本文献已被 万方数据 等数据库收录！ |
