This module torch.nn also has various layers that you can use to build your ... PyTorch that we want the softmax computed over the right-most dimension.
確定! 回上一頁