Implement Softmax Function

Given a 2D tensor of raw scores (logits) where each row represents a distinct data point and each column represents a class, implement the softmax function to transform the raw scores into probabilities. Ensure that the probabilities for each data point sum to 1. Return the resulting tensor of probabilities.

The softmax function is often used in the output layer of a neural network model to represent the probability distribution of the possible classes. Check hint for specifit steps.