Image Classification using Logistic Regression in PyTorch

# Imports
import torch
import torchvision
from torchvision.datasets import MNIST

from torch.utils.data import random_split

import matplotlib.pyplot as plt
%matplotlib inline

import torchvision.transforms as transforms