5fa1a76
1
2
3
Tips: VAN does not have an embedding layer, thus the hidden_states will have a length equal to the number of stages.