[일반] pytorch LSTM 입력 차원 질문

[ㅁㅁㅁㅁ(157.254)]

[seq_len, batch_size, input_size]

vs

[batch_size, seq_len, input_size]

뭐가 맞는거임?

[익명(222.107)]

batch_first=False면 전자 True면 후자 트랜스포머도 동일할거고

[익명(222.107)]

https://pytorch.org/docs/stable/generated/torch.nn.LSTM.html

[ㅁㅁㅁㅁ(23.26)]

아 저 파라미터가 저 차이였냐 ㅅㅂ