Question d’entretien chez Algolux

Write a pytorch code for a simple problem