[英]How to construct the following matrix elegantly in numpy?
Suppose I have a 5 dimensional matrix v
and now I want a new matrix D
fulfilling 假设我有一个5维矩阵
v
,现在我想要一个新的矩阵D
D[a, b, n, m, d] = v[a, b, n, n, d]-v[a, b, m, m, d].
How do I elegantly do this in numpy? 我如何用numpy优雅地做到这一点?
How do you want to change the dimensionality? 您想如何更改尺寸? You can reshape it like this
您可以像这样重塑它
import numpy as np
a, b, n, d = 2, 3, 4, 5
v = np.zeros((a, b, n, n, d))
D = v.reshape((a, b, n*n, d))
我发现einsum
可以做到这一点:
D = np.einsum('abiic->abic', v)[..., None, :] - np.einsum('abiic->abic', v)[:, :, None, ...]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.