技术背景 Vmap是一种在python里面经常提到的向量化运算的功能,比如之前大家常用的就是numba和jax中的向量化运算的接口.虽然numpy中也使用到了向量化的运算,比如计算两个numpy数组的加和,就是一种向量化的运算.但是在numpy中模块封装的较好,定制化程度低,但是使用便捷,只需要调用最上层的接口即可.现在最新版本的mindspore也已经推出了vmap的功能,像mindspore.numba还有jax,与numpy的最大区别就是,需要在使用过程中对需要向量化运算的函数额外嵌套一层…