Pytorch classes

Create class SQ with custom backward and forward. Forward should take i and square it.

Create class toy_set. The set should be 100 by default, with no transformers. x should be (2, 2) for all examples, and y should be 1. x is on index 1, and y is on index 2.

Create class add_mult. It add 1 to x and multiplies y by 2. The data is contained in sample. x is the 1st index, and y is the 2nd index. Return sample as x, y

pass

pass