mxnet gluon GRU 文档

作者：知新_RL | 2024-04-27 08:52:53

踩

mxnet gluon GRU 文档

mxnet.gluon.rnn.GRU官方文档
以下是一个使用的简单用例，详细信息前往官网

# hidden_size = 100 num_layer = 3
layer = mx.gluon.rnn.GRU(100, 3)
layer.initialize()
# seq_len = 5 batch_size = 3 input_size = 10
input = mx.nd.random.uniform(shape=(5, 3, 10))
# by default zeros are used as begin state
output = layer(input)
# manually specify begin state.
# num_layers = 6 batch_size = 3 num_hidden = 128
h0 = mx.nd.random.uniform(shape=(3, 3, 100))
output, hn = layer(input, h0)
1
2
3
4
5
6
7
8
9
10
11

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/知新_RL/article/detail/495864