vltnmmdv's picture
return view(-1, 1) in infer mode (#2)
d676181 verified