Compute a * (b %*% one_mat) where a, b,
ones_mat are square matrices of the same size,
and ones_mat contains all entries equal to one.
The product * is an entry-wise (Hadamard) product,
while %*% represents matrix multiplication.
This method is more efficient than the naive approach
when a or b are sparse.