Can't compile the student's code. RuntimeError

I think this is another case in which the torch version that the grader is using is less permissive than the one in the notebook. It looks like the grader’s version of can’t handle different types. I wrote that code like this and it passes the grader:

combined =, y.float()), dim = 1)

The point is that I do the conversion to float before the cat, of course.