Document sizes of backward functions

[srush]

• If I understand correctly, should this return be enforced to be something with the shape of a? https://github.com/minitorch/Module-2/blob/master/minitorch/tensor.py#L339