资源说明:通过源代码中可以看到nn.LSTM继承自nn.RNNBase,其初始化函数定义如下
class RNNBase(Module):
...
def __init__(self, mode, input_size, hidden_size,
num_layers=1, bias=True, batch_first=False,
dropout=0., bidirectional=False):
我们需要关注的参数以及其含义解释如下:
input_size – 输入数据的大小,也就是前面例子中每个单词向量的
本源码包内暂不包含可直接显示的源代码文件,请下载源码包。