开发者论坛

全部回复

LV.1 #7 hellokitty回复
#6lichaoqian回复
是的
1.tf2中是可以通过keras来调用lstm算子，您可以在网上查阅上社区tf2中如何通过keras调用lstm；
2.参考tf2的sdk手册，5.3节，查看如何在370上使用keras
2025-05-27 09:32:31 0| 回复
LV.1 #6 lichaoqian回复
#5hellokitty回复
您的需求是什么呢？在tf2中调用lstm算子吗？
是的
2025-05-27 09:19:11 0| 回复

LV.1 #5 hellokitty回复

#4lichaoqian回复

CNNL1.28.6文档里面有一个新的函数cnnlRNNForwardInference_v2。文档中说跟cnnlRNNForwardInference的区别是w_desc, b_desc的类型变了。但是详细描述还是跟cnnlRNNForwardInference一样的。我不知道w_desc, b_desc该怎么初始化。同时cnnlInitRNNExtraInput也没有办法使用，因为w_desc,b_desc类型不同，没有办法设置extra_input。rnn_sample里面的代码用的是cnnlRNNForwardInference。所以说cnnlRNNForwardInference_v2现在还不可用？TF1的代码里面CambLSTM用的是cnnlRNNForwardTraining，这个更简单，调用还少了一次MemcpyH2D。那么推理应该用这个吗？在帮看下这两个问题

展开

您的需求是什么呢？在tf2中调用lstm算子吗？

2025-05-26 16:52:36 0| 回复

LV.1 #4 lichaoqian回复
#3hellokitty回复
“在tf2中，请参考使用社区版本中LSTM的调用方式”是指Tensorflow社区版本的调用方式，参考tensorflow2的标准用法
展开
CNNL1.28.6文档里面有一个新的函数cnnlRNNForwardInference_v2。文档中说跟cnnlRNNForwardInference的区别是w_desc, b_desc的类型变了。但是详细描述还是跟cnnlRNNForwardInference一样的。
我不知道w_desc, b_desc该怎么初始化。同时cnnlInitRNNExtraInput也没有办法使用，因为w_desc,b_desc类型不同，没有办法设置extra_input。
rnn_sample里面的代码用的是cnnlRNNForwardInference。所以说cnnlRNNForwardInference_v2现在还不可用？

TF1的代码里面CambLSTM用的是cnnlRNNForwardTraining，这个更简单，调用还少了一次MemcpyH2D。那么推理应该用这个吗？

在帮看下这两个问题
2025-05-26 10:31:12 0| 回复
LV.1 #3 hellokitty回复
#2lichaoqian回复
你好，第二点能帮发下链接？在论坛中没有搜索到
“在tf2中，请参考使用社区版本中LSTM的调用方式”
是指Tensorflow社区版本的调用方式，参考tensorflow2的标准用法
2025-05-26 08:42:04 0| 回复
LV.1 #2 lichaoqian回复
#1hellokitty回复
您好，以下请参考：1.CamLSTM在tf2中移除；2.在tf2中，请参考使用社区版本中LSTM的调用方式
你好，第二点能帮发下链接？在论坛中没有搜索到
2025-05-22 16:09:50 0| 回复
LV.1 #1 hellokitty回复

您好，以下请参考：
1.CamLSTM在tf2中移除；
2.在tf2中，请参考使用社区版本中LSTM的调用方式

2025-05-22 15:20:24 0| 回复

请登录后评论