Numpy elementwise product of 3d array,"np.einsum('ijk,ikl->ijl', A, B)"