Hi! I have two matrices for which I need to calculate the dot product, but only for one dimension. They are of the same shape (N,M,D) and I want to take the dot product of the last dimension D so that the result is of shape (N,M,1). How would I do that? Most solutions seem to assume stuff about number of dimensions etc.