Solved by ptrblck in post #2 
              
              
                I don’t know what the TF code is exactly doing, but in PyTorch you can save and load numpy arrays as well and adding them to a dict is also a valid idea: 
a= np.random.randn(10)
checkpoint = {
    'np_arr': a
}
torch.save(checkpoint, 'tmp.pth')

b = torch.load('tmp.pth')['np_arr']

print(all(a == b)…