20. Replace Max with 0

Medium

Replace the maximum value of tensor a with 0, leaving everything else unchanged. Return the result.

Implement solve(...)