pytorch

decec8f5 - Implement Distribution.expand() and use it in a few places