python 声音模仿_python – 我如何做一个模仿’keepdims’的einsum？

一个python问题：我有一个np.einsum操作,我正在一对3d数组上做：

return np.einsum('ijk, ijk -> ik', input_array, self._beta_array)

我遇到的问题是结果是2d;该操作折叠’j’维度.我喜欢做的是让它保留’j’维度,类似于’keepdims’在np.sum函数中的工作方式.

我可以将结果包装在np.expand_dims中,但这对我来说似乎效率低下.我宁愿找到一些方法来调整einsum来输出我想要的东西.

这是真的吗？

解决方法:

I can wrap the result in np.expand_dims, but that seems inefficient to me

在numpy中添加维度是最坏的O(ndim),所以基本上是免费的.至关重要的是,实际数据没有被触及 – 所有发生的事情都是.strides和.shape元组各得到一个额外的元素

现在没办法用einsum直接得到你想要的东西.

您可以尝试针对numpy提出拉取请求,以支持像ijk,ijk – >这样的内容. i1k,如果你真的认为它提高了可读性

标签：python,numpy,numpy-einsum

来源： https://codeday.me/bug/20190828/1753072.html