Element wise multiplication of the last dimension

[googlebot (Alex)]

You don’t need expand() there, tensor broadcasting takes care of it.