标签： python arrays indexing profiling
通过逐行分析我的代码，我观察到我的程序的瓶颈，是以下命令：
A[index_array]
A 和 index_array 是2D矩阵。结果是3D阵列。现在，在程序中使用了40％的时间，该程序还使用了昂贵的命令，如np.einsum和Softmax函数。
我是python编码的新手，我无法找到更有效，更快速的编写方法。有人可以提出建议吗？