Add a pytorch adapter similar to the keras one.
Add a pytorch adapter similar to the keras one.