Warning: file_get_contents(/data/phpspider/zhask/data//catemap/2/tensorflow/5.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
Tensorflow torch.nn.LSTM是否等于tf.nn.rnn_cell.BasicLSTMCell?_Tensorflow_Pytorch_Lstm - Fatal编程技术网

Tensorflow torch.nn.LSTM是否等于tf.nn.rnn_cell.BasicLSTMCell?

Tensorflow torch.nn.LSTM是否等于tf.nn.rnn_cell.BasicLSTMCell?,tensorflow,pytorch,lstm,Tensorflow,Pytorch,Lstm,一篇论文为分类任务发布了tensorflow代码。它依赖于默认设置的tf.nn.rnn_cell.BasicLSTMCell 我尝试使用PyTorch 1.0重新实现,并使用torch.nn.LSTM和默认设置。但测试精度低于tensorflow规范 我检查了这两个lstm的使用情况,我认为默认设置是相同的。所以我很困惑 我认为torch.nn.LSTM相当于tf.contrib.cudnn\u rnn.CudnnLSTM,而不是tf.nn.rnn\u cell.BasicLSTMCell。我对

一篇论文为分类任务发布了tensorflow代码。它依赖于默认设置的tf.nn.rnn_cell.BasicLSTMCell

我尝试使用PyTorch 1.0重新实现,并使用torch.nn.LSTM和默认设置。但测试精度低于tensorflow规范


我检查了这两个lstm的使用情况,我认为默认设置是相同的。所以我很困惑

我认为
torch.nn.LSTM
相当于
tf.contrib.cudnn\u rnn.CudnnLSTM
,而不是
tf.nn.rnn\u cell.BasicLSTMCell
。我对tensorflow还不太熟悉,不能肯定,但我希望其他人能确认并将其作为答案发布。@myrtlecat是正确的,如果您运行的是gpu@myrtlecat谢谢您能再解释一下CPU的区别吗?谢谢。还有CPU的区别