input_shape (tuple of int) – Input tensor shape at each time step for each sample, excluding dimension of the batch size
and sequence length. Must be consistent with conv_layout.
For example, for layout ‘NCHW’ the shape should be (C, H, W).

activation (str or Block, default 'tanh') – Type of activation function used in n_t.
If argument type is string, it’s equivalent to nn.Activation(act_type=str). See
Activation() for available choices.
Alternatively, other activation blocks such as nn.LeakyReLU can be used.

prefix (str, default 'conv_gru_’) – Prefix for name of layers (and name of weight if params is None).