... 转一下本篇主要分析softmax函数中的dim参数, 首先介绍一下softmax函数:设则接下来分析torch.nn里面的softmax函数y = torch.tensor([[[1.,2.,3.]
確定! 回上一頁