HF RL Explorer

Task: PyTorch Softmax Dimension Understanding

Task: PyTorch Softmax Dimension Understanding: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that demonstrates the correct usage of torch.nn.functional.softmax with respect to the dim parameter. Specifically, prove that when you want the sum along a particular dimension to…

The task

Write a Python script that demonstrates the correct usage of `torch.nn.functional.softmax` with respect to the `dim` parameter. Specifically, prove that when you want the sum along a particular dimension to equal 1 after applying softmax, you must apply softmax along that same dimension.

Part of PrimeIntellect/Terminal-Lego-15k.