We present trainsum, a versatile Python package for doing computations with multidimensional quantics tensor trains: https://github.com/fh-igd-iet/trainsum. Using the Array API standard together with opt_einsum, trainsum allows the effortless approximation of tensors or functions by tensor trains independent of their shape or dimensionality. Once approximated, our package can perform normal arithmetic operations with quantics tensor trains, including addition, Einstein summations and element-wise transformations. It can be therefore used for generic computations with applications in simulation, data compression, machine learning and data analysis.
翻译:本文介绍trainsum,一个用于多维量子化张量列车计算的通用Python包:https://github.com/fh-igd-iet/trainsum。该包基于Array API标准并结合opt_einsum,能够对任意形状或维度的张量及函数进行张量列车近似。完成近似后,本包可对量子化张量列车执行常规算术运算,包括加法、爱因斯坦求和及逐元素变换。因此,该工具可广泛应用于模拟、数据压缩、机器学习及数据分析等领域的通用计算任务。