fix input to softmax_to_unary

[peteflorence]

thanks for this helpful ipy notebook with the crf example!

this tripped me up, but was a simple fix. thought might be helpful for others.

i also have a version of this running on 1.2, if interested in having me PR that