add raw_logits in categorical to() method (#131)

add raw_logits in categorical to() method (#131) #818