Size([2, 3, 5]) print(x.permute(2, 0, 1).size()) >>>torch.Size([5, 2, 3]) LogSoftmax: Log ... top_value, top_index = torch.topk(x, 3) print(top_value, ...
確定! 回上一頁