For example, I have many nn.linear layer, and I need to input the same feature into different nn.linear, currently, the solution is as follows:
for i in range(n):
output = nn.Linear()(feature)
So, how can I improve the computational efficiency? If possible, please give me some suggestions? Thanks a lot!