gru参数数量为0怎么回事?(pytorch)

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))

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

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

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