Pregunta de entrevista de Algolux

Write a pytorch code for a simple problem