You mean input this tensor as parameter for the model instance? But I do not think it is a ideal way to solve this problem.
I tried to first register this tensor to model attribute, like:
model.A = A

and use it during model’s forward method, but the error raised.