So in this case, this tells PyTorch that we want to take the mean ranging over the values indexed by the last two axes of the tensor.
確定! 回上一頁