2

私たちが持っているとき:

array 1: A, shape (49998,3,3) 
array 2: B, shape (3, 49998)

そして、それらの部分配列を掛けて取得したい

array 3: C, shape(3,49998)

ジェネレーターを使用しているim:

def genC(A,B):
    for a,b in itertools.izip(A,B.T):
        c=np.dot(a,b)
        yield c.T[0]

C=np.array([c for c in genC()]).T

では、配列 C を取得するために for ループを使用せずに、A、B の内部で配列乗算を行うにはどうすればよいでしょうか。

使おうと思ったnp.tensordotけど取れない

ノート:

これは、私が持っていた元のデータのいくつかのテストcosの基本的な例です

4*3*37 配列 A(500 000,3,3) B(3,500 000)

to do、および for loop sems は、pythonic の方法ではありません xD

4

1 に答える 1