MemSaveConvTranspose1d¶
- class memsave_torch.nn.MemSaveConvTranspose1d(in_channels: int, out_channels: int, kernel_size: int | Tuple[int], stride: int | Tuple[int] = 1, padding: int | Tuple[int] = 0, output_padding: int | Tuple[int] = 0, groups: int = 1, bias: bool = True, dilation: int | Tuple[int] = 1, padding_mode: str = 'zeros', device=None, dtype=None)¶
Differentiability-agnostic 1d transpose convolution layer.
Initializes internal Module state, shared by both nn.Module and ScriptModule.
- forward(input: Tensor) Tensor¶
Forward pass.
- Parameters:
input (torch.Tensor) – Input to the network [B, C_in, W]
- Returns:
Output [B, C_out, W_out]
- Return type:
- classmethod from_nn_ConvTranspose1d(convT1d: ConvTranspose1d)¶
Converts a nn.ConvTranspose1d layer to MemSaveConvTranspose1d.
- Parameters:
convT1d (nn.ConvTranspose1d) – The nn.ConvTranspose1d layer
- Returns:
The MemSaveConvTranspose1d object
- Return type:
Hint
The usage is the same as torch.nn.ConvTranspose1d
For usage examples, please refer to the linked torch documentation