https://researchdatapod.com/how-to-convert-numpy-array-to-pytorch-tensor/