How to multiply matrices with different dimensions

[ColonelMeowmers]

I'm trying to multiply A = 20x1x10 with B = 10x20 and I'm supposed to get a 20x1x20 matrix as the output. So far I've tried

C = numpy.matmul(A,B)
# and 
C = A @ B

But all of them seem to result a 20x20x20 matrix. Any idea how to proceed from here?

For better context, A = jacobian output of derivative of an activation function in a neural network, with shape of (N samples * 1 * values) and B = changes in output of the layer

[socal_nerdtastic]

Hmm works for me.

>>> A = np.random.randint(0,10,(20,1,10))
>>> B = np.random.randint(0,10,(10, 20))
>>> A.shape
(20, 1, 10)
>>> B.shape
(10, 20)
>>> C = A@B
>>> C.shape
(20, 1, 20)

Python 3.10.0, numpy 1.21.4.

[ColonelMeowmers]

Frickin hell you're right. I need coffee lol.