Pytorch CNN input dimensions not matching

welcome to shbcf.ru