Replace the maximum value of tensor a with 0, leaving everything else unchanged. Return the result.
a
Sign in to write and run your own code.
solve(...)