Solved by SimonW in post #2 
              
              
                The transform you referred to is available at master, but not in a release yet. You can build from source (https://github.com/pytorch/vision). It should be very easy.