How to understand aten::lsm

1

Hi , I can’t understand about the args of aten::lstm function

How to match the args of aten::lstm and torch.nn.LSTM()

Thanks.

ps:I am not found the api doucment about aten::lstm.

Library API — PyTorch master documentation

2

You can specify the options via LSTMOptions and create an LSTM layer via e.g.:

auto lstm_opt = LSTMOptions(1, 1).num_layers(1).batch_first(false).bidirectional(false);
LSTM lstm{lstm_opt);
3

oh, awsome, thanks very much