* Add parameter to `scatter_[log_]softmax` * Update softmax.py * Format code.
Attach a file by drag & drop or click to upload
