develop分支能否支持单向GRU呢
Created by: richlaji
我看到paddle实现双向GRU的方式是:
fc_1 = fluid.layers.fc(input=x,
size=rnn_hidden_size * 3,
param_attr=para_attr,
bias_attr=bias_attr_nobias)
fc_2 = fluid.layers.fc(input=x,
size=rnn_hidden_size * 3,
param_attr=para_attr,
bias_attr=bias_attr_nobias)
gru_forward = fluid.layers.dynamic_gru(
input=fc_1,
size=rnn_hidden_size,
param_attr=para_attr,
bias_attr=bias_attr,
candidate_activation='relu')
gru_backward = fluid.layers.dynamic_gru(
input=fc_2,
size=rnn_hidden_size,
is_reverse=True,
param_attr=para_attr,
bias_attr=bias_attr,
candidate_activation='relu')
那对于单向的Pytorch上的代码是:
model = nn.GRU(10, 20, 2)
input = torch.randn(5, 3, 10)
#h0 = torch.zeros(2, 3, 20)
yp = model(input)
torch.onnx.export(model, input, 'gru.onnx',verbose=True)
转后的onnx模型是 链接:https://pan.baidu.com/s/1NtYNN5DSALo-tLrGFDlGxA 提取码:897a
想问一下目前develop分支能否支持单向GRU