PyTorch internals : Inside 245-5D