to-loadable-in-cpp.py

import torch
from torch import nn

class TensorModule(nn.Module):
    def __init__(self, tensors):
        super(TensorModule, self).__init__()

        for tensor_name, tensor in tensors.items():
            setattr(self, tensor_name, nn.Parameter(tensor))

# Then, given some string -> Tensor dictionary.
wrapper = TensorModule(tensor_dict)
script = torch.jit.script(wrapper)
script.save(args.tensors)