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個特徵。由於組的大小不同,實際上,它是不同長度的列表的列表,具有相同長度的列表。 有一個規則的結構,因此可以保存爲一個標準