https://onlinebme.com/pytorch-vs-tensorflow-why-i-choose-pytorch/