rnn.py 文件源码

def init_rnn_descriptor(fn):
    rnn_desc = cudnn.RNNDescriptor()

    rnn_desc.set(
        fn.hidden_size,
        fn.num_layers,
        fn.dropout_desc,
        fn.input_mode,
        fn.bidirectional,
        fn.mode,
        fn.datatype
    )
    return rnn_desc