Newbie in PyTorch (&DL) and trying to figure things out. bear with me