Softmax函数常用的用法是指定参数dim 就可以:. tensor([2, 3]))a[[2, ... gather (get elements by looking up the table), scatter_] (get elements by looking .
確定! 回上一頁