pt2onnx only get the float32 type output

[peanutpaste]

I use pytorch2onnx.py file to convert .pt to .onnx, i want get the output of onnx type is float32,but got int64,how can i get float32,thx!