Numpy Array Vs Pytorch Tensorboard Files