RNN — PyTorch 1.10.1 documentation
pytorch.org › docs › stableRNN. class torch.nn.RNN(*args, **kwargs) [source] Applies a multi-layer Elman RNN with. tanh . \tanh tanh or. ReLU. \text {ReLU} ReLU non-linearity to an input sequence. For each element in the input sequence, each layer computes the following function: h t = tanh ( W i h x t + b i h + W h h h ( t − 1) + b h h)
torch.nn — PyTorch 1.10.1 documentation
pytorch.org › docs › stablenn.ConvTranspose3d. Applies a 3D transposed convolution operator over an input image composed of several input planes. nn.LazyConv1d. A torch.nn.Conv1d module with lazy initialization of the in_channels argument of the Conv1d that is inferred from the input.size (1). nn.LazyConv2d.