Turn Numpy Array Into Tensorboard Pytorch