I have a model that takes a Python class as a parameter to forward():
class Inputs:
def __init__(self, a:Tensor, b:Tensor, c:Tensor):
self.a = a
self.b = b
self.c = c
class Model(nn.Module):
def forward(self, inputs: Inputs):
...
After converting it to a ScriptedModule and saving to a file, I would like to load and use this model in a C++ program.
However, I am not sure what would be the equivalent of Inputs
class on the C++ side… Any pointers to docs/examples that show how to do this?