Solved by goldsborough in post #2 
              
              
                You can get the raw integer by calling output.item<int64_t>(). 
output.item() (without template arguments) returns a torch::Scalar and might also be useful output.