Python 3.x, Pytorch 2.x compatability fixes

[prasad83]

Python 3.x changes

• print "ABC" to print ("ABC")
• [] + map( ... ) to [] + list(map( .... ) )

Pytorch 2.x changes

• hidden_size should be int instead of float
• tensor.numpy() to tensor.numpy().flatten()