pytorch - pytorch 中 torch.nn.gru 函数的输入是什么？

Question

我正在使用 gru 函数来实现 RNN。这个 RNN (GRU) 在一些 CNN 层之后使用。有人可以告诉我这里的 GRU 函数的输入是什么吗？特别是，隐藏大小是否固定？

self.gru = torch.nn.GRU(
            input_size=input_size,
            hidden_size=128,
            num_layers=1,
            batch_first=True,
            bidirectional=True) 

根据我的理解，输入大小将是特征数量，而 GRU 的隐藏大小始终固定为 128？有人可以纠正我。或者给他们反馈

Answer 1

一、GRU不是一个函数而是一个类，你正在调用它的构造函数。您正在创建类 GRU 的实例在这里，这是一个层(或 Module 在 pytorch 中)。input_size必须匹配 out_channels之前的 CNN 层。
您看到的所有参数都不是固定的。只需在那里放另一个值，它就会是别的东西，即用你喜欢的任何东西替换 128。
即使它被称为 hidden_size ，对于 GRU，这个参数也决定了输出特征。换句话说，如果您在 GRU 之后还有另一层，则该层的 input_size (或 in_features 或 in_channels 或任何名称)必须匹配 GRU 的 hidden_size .
另外，看看 documentation .这会告诉您传递给构造函数的参数的确切用途。此外，它会告诉您实际使用图层后的预期输入(通过 self.gru(...) )以及该调用的输出是什么。