oh, and .view(-1, ...) means to squish an n-dimensional vector so that is has the dimension sizes listed, where -1 means to make that dimension as large as needed to fit all the needed. so .view(-1) turns it into a 1-dimensional array.