How can I implement a dynamic parameter in Linear Layer by using pytorch?

[JosueCom (Josue)]

I implemented your suggestion but the output shape does not change

Dynamic linear layer doesn't change output when weights shape is changed

If I implement a dynamic layer as present below, it’s output shape doesn’t get affected. See code below for greater details. >>> import torch as th >>> import torch.nn as nn >>> import torch.nn.functional as F >>> class DyLinear(nn.Module): ... def __init__(self, in_fea, out_fea, device = 'cuda'): ... super(DyLinear, self).__init__() ... self.device = device ... self.in_fea = in_fea ... self.out_fea = out_fea ... self.weight = nn.Parameter(th.randn(ou…