7.1 Built-in modules
- What are modules?
-
- an object that encapsulates state
- can be of any complexity (e.g. layer, or models consisting of layers)
Examples of {torch} modules:
- linear:
nn_linear() - convolutional:
nn_conf1d(),nn_conf2d(),nn_conv_3d() - recurrent:
nn_lstm(),nn_gru() - embedding:
nn_embedding() - multi-head attention:
nn_multihead_attention() - See torch documentation for others