a4566752a9207c7769e00c97e08f905d601f27e9,pymanopt/tools/multi.py,,multiprod,#,5

Before Change

    if len(np.shape(A)) == 2:
        return np.dot(A, B)

    a = A.reshape(np.hstack([np.shape(A), [1]]))
    b = B.reshape(np.hstack([[np.shape(B)[0]], [1], np.shape(B)[1:]]))
    return np.sum(a * b, axis=2)


def multitransp(A):
    

After Change

    //return np.sum(a * b, axis=2)

    // Approx 5x faster, only supported by numpy version >= 1.6:
    return np.einsum("ijk,ikl->ijl", A, B)


def multitransp(A):
    

Instances