摘要: # save x = torch.tensor([1,2,3]) np.savez("./tmp.npz",x) # save dict save_inp_file = "./text.npz" a = [torch.tensor([i]) for i in range(5)] b = [torch 阅读全文
posted @ 2022-01-10 17:45 MrDoghead 阅读(491) 评论(0) 推荐(0) 编辑