Solved by desert_ranger in post #2 
              
              
                Figured it out - 
import numpy as np
from sklearn.datasets import load_sample_images
import matplotlib.pyplot as plt
import torch
import torch.nn as nn
import torchvision.transforms as transforms

def random_crop(imgs, out=84):


    imgs = torch.tensor(imgs)
    change = torch.nn.Sequential(
      …