Torch resize tensor. unsqueeze(0) the input and .