先看看文檔怎么說的

這里說指定某個(gè)緯度（dim)，就會(huì)讓那個(gè)緯度的和為1

看一下input是2維的情況

dim=0

dim=1

再看看input是3維的情況

dim = 0

dim = 1