如何使用Cudnn LSTMP？

Question

通常，我使用tf.contrib.cudnn_rnn.CudnnLSTM（张量流1.15），现在想使用LSTMP（LSTM投影）。我知道tf.contrib.rnn.LSTMCell具有LSTMP，但是额外的培训时间对我来说太长了。我发现了nvidia support LSTMP。如何在tensorflow（或其他ML工具包）上使用它？

谢谢！

Answer 1

[您也可以使用tf.contrib.cudnn_rnn.CudnnParamsFormatConverterLSTM，其中CudnnParamsFormatConverterLSTM是在Cudnn和TF LSTM参数之间转换的帮助程序类。您也可以使用num_proj设置投影矩阵的输出尺寸。这是TensorFlow 1.15中API的详细文档的link。