疯狂的矩阵、张量加速函数！

文章预览

简介 在平时的工作中，包括时间序列等的建模中，我们常常会涉及到线性或者Multilinear代数的计算，很多时候包括我在内也喜欢偷懒使用numpy来处理，个人觉得还是可以的，但有的朋友觉得这个看起来可读性会差一些，并且速度上也差一些，今天就推荐一个新的函数 einsum 函数。 通过 einsum 函数，我们可以写出可读性更好，更加高效的函数。 einsum函数 01 关于einsum函数 einsum函数一般会用于向量，矩阵，张量的计算中。 如果输入的是标量（scalar），我们就可以空着，因为它没有下标。 02 einsum函数优势 1.速度优势 因为numpy很多是用C写的。 如果我们使用Python循环时，所有数据操作都在Python解释器中进行。 当使用内置numpy函数时，它发生在C中，这就是numpy速度更快的原因。 使用einsum时，numpy会在C中处理数据一次并返回最终结果，而使用多个numpy函数会 ………………………………