gru参数数量为0怎么回事？（pytorch）

Question

class GRUNet(nn.Module):
    def __init__(self,input_size,hidden_size,n_layers,output_size):
        super(GRUNet,self).__init__()
        self.input_size = input_size
        self.hidden_size = hidden_size
        self.n_layers = n_layers
        self.output_size = output_size
        
        self.gru = nn.GRU(input_size,hidden_size,n_layers,batch_first=True)
        self.fc1 = nn.Sequential(
            nn.Linear(hidden_size,output_size)
            )
    def forward(self,x):
        r_out,h_n = self.gru(x,None)## None 表示初始值的hidden_state为0(gru中有两个参数，一个是输入，一个是hidden)
        out = self.fc1(r_out[:,-1,:])
        return out
grumodel = GRUNet(8,20,3,1).to(device)
summary(grumodel, (96,8))

为什么统计的gru参数数量为0

Answer 1

你好，我是有问必答小助手。为了技术专家团更好地为您解答问题，烦请您补充下（1）问题背景详情，（2）您想解决的具体问题，（3）问题相关代码图片或者报错信息。便于技术专家团更好地理解问题，并给出解决方案。

您可以点击问题下方的【编辑】，进行补充修改问题。