util_chainer.py 文件源码

def my_rnn_link(rnn_link, n_layers, feature_dim, hidden_dim, use_dropout, use_cudnn):
    if version < '2.0':
        return rnn_link(n_layers=n_layers, in_size=feature_dim,
                        out_size=hidden_dim, dropout=use_dropout,
                        use_cudnn=use_cudnn)
    else:
        # v2.0
        return rnn_link(n_layers=n_layers, in_size=feature_dim,
                        out_size=hidden_dim, dropout=use_dropout)