PyTorch: softmax dim

F.softmax needs a dim argument specifying the:

Sign in to solve this challenge

Use your college email to start solving, earn XP, build a streak, and climb your college leaderboard. It is free.