About Dropout

[JohnMasoner]

Hey,

Many thanks to you guys for providing this pypi package.

When I used this library, I found that the optional parameter of Dropout is Union[None, Type[_DropoutNd]], dropout_op_kwargs is dict.

But when I use this library, I need to set different parameters between different layers.

Could the dropout_op and dropout_op_kwargs support List?