numpy-einsum

    3热度

    1回答

    目前我使用 Na = (3, 2, 4) Nb = Na[1:] A = np.arange(np.prod(Na)).reshape(Na) b = np.arange(np.prod(Nb)).reshape(Nb) 我想要计算: r = np.empty((A.shape[0], A.shape[2]) for i in range(A.shape[2]): r[:,

    2热度

    1回答

    我有一批的存储在数组中xbm x n图像,和卷积滤波器尺寸p x q,我想应用到每个图像的f在批处理(然后使用总和池和存储在一个阵列y),即all(np.allclose(y[i][j][k], (x[i, j:j+p, k:k+q] * f).sum()) for i in range(b) for j in range(m-p+1) for k in range(n-q+1))为真。 适应th

    3热度

    1回答

    我想了解numpy.einsum()函数,但文档以及从stackoverflow this answer仍留下一些问题。 我们来看看爱因斯坦和和答案中定义的矩阵。 A = np.array([0, 1, 2]) B = np.array([[ 0, 1, 2, 3], [ 4, 5, 6, 7], [ 8, 9, 10, 11]]) np.einsum(

    1热度

    1回答

    更快的矩阵运算更换顺序积和在我目前的theano脚本的瓶颈是下面的代码: import numpy as np axis = 0 prob = np.random.random((1, 1000, 50)) cases = np.random.random((1000, 1000, 50)) start = time.time() for i in xrange(1000):

    2热度

    2回答

    我正在尝试基于矩阵编写快速优化的代码,并且最近发现了einsum作为实现显着加速的工具。 是否可以使用它来有效地设置多维数组的对角线,还是只能返回数据? 在我的问题中,我试图通过求和每个正方形(N×N)矩阵中的列来设置矩阵矩阵(形状:M x N x N)的对角线。 我现在的(基于循环慢,)的解决方案是: # Build dummy array dimx = 2 # Dimension x (li

    4热度

    1回答

    我想在python中实现高斯混合模型的期望最大化算法。 我有以下行来计算高斯概率p我的数据X给出的平均亩和协方差高斯分布的西格玛的: for i in range(len(X[0])): p[i] = scipy.stats.multivariate_normal.pdf(X[:,i],mu,sigma) 我想知道我以某种方式可以摆脱for循环得到像 p[:] = scipy.sta

    2热度

    1回答

    在tf.einsum中可以使用多于26个小写字母作为索引吗? Numpy还允许使用大写字母,即np.einsum('zA,AB->zB',M1,M2)而tf.einsum会返回错误。 这对收缩张量网络非常有用。

    0热度

    3回答

    我有两个ndarray像 n1 = np.array([1,2,3,4]) n2 = np.array([1,2,3,4]) 虽然它们的点积可以用np.dot(n1, n2),这给30正确的答案很容易做到。如果我需要在n1和n2的两个子阵列上操作该点,例如 np.dot(np.array([1,2]), np.array([1,2])) # first two elements from a

    1热度

    2回答

    给出一个三维阵列和二维阵列的最终轴, a = np.arange(10*4*3).reshape((10,4,3)) b = np.arange(30).reshape((10,3)) 如何运行跨过每个的最终轴的elementwise乘法,导致c其中c具有形状.shape作为a?即 c[0] = a[0] * b[0] c[1] = a[1] * b[1] # ... c[i] = a

    3热度

    1回答

    我想执行操作 如果有一个规则的形状,然后我可以使用np.einsum,我相信语法将是 np.einsum('ijp,ipk->ijk',X, alpha) 不幸的是,我的数据X有一个非规则的结构(如果我们是零指数)的轴。 为了提供更多的背景知识,指的是第i组第j个成员的第p个特征。由于组的大小不同,实际上,它是不同长度的列表的列表,具有相同长度的列表。 有一个规则的结构,因此可以保存为一个标准