Softmax(dim=1) x_input = torch.randn(1, 3) y_output = softmax(x_input) print(x_input) print(y_output) print(torch.sum(y_output, dim=1)) Output[0] tensor([[ ...
確定! 回上一頁