Hi! I posted the question only for asking for sparse operation. so to_dense is not what I want.
The multiplication behavior is the same as A @ B, which multiplies the 3th and 4th dimension and broadcast in the 1st dimension.