Softmax函数常用的用法是指定参数dim 就可以:. size ()[0] # batch_size ... 05) and the corresponding targets from the training data would be like (0, 1, 0).
確定! 回上一頁