So for instance, in the example above, the model expects a tensor as
input, since it is the input to the module a, and its output will is a
So for instance, in the example above, the model expects a tensor as
input, since it is the input to the module a, and its output will is a