Change softmax and log_softmax to take int64_t dim rather than int.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
1 file changed