pytorch之expand，gather，squeeze，sum，contiguous，softmax，max，argmax