embedding_only_onnx.py

import torch.onnx
from torch.autograd import Variable
import torch.nn as nn

inp = Variable(torch.LongTensor([1,2,3]))

class MyModel(nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
        self.embedding = nn.Embedding(100, 300)
    def forward(self, x):
        out = self.embedding(x)
        return out

model = MyModel()
out = model(inp)

torch.onnx.export(model, inp, 'embedding_only.onnx', verbose=True)