Meta interview question

Write a pytorch function achieving 2D Gaussian filtering and optimization