张量运算的类型安全抽象
编程语言
2017-10-20 v1
摘要
我们提出了一种针对张量(即多维数组)的类型安全抽象,利用 Scala 通过异构列表(HList)实现的类型级编程能力,并展示了常见张量运算及各种神经层(如卷积或循环神经网络)的类型安全抽象。这种抽象可为未来运行于 Scala/JVM 上的类型安全深度学习框架奠定基础。
引用
@article{arxiv.1710.06892,
title = {Typesafe Abstractions for Tensor Operations},
author = {Tongfei Chen},
journal= {arXiv preprint arXiv:1710.06892},
year = {2017}
}