This is a part of issue #2628, we will patch meta information of `torch.nn.functional.softmax()`
This is a part of issue #2628, we will patch meta information of
torch.nn.functional.softmax()