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:

torch.Tensor

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:

MemSaveConvTranspose1d

Hint

The usage is the same as torch.nn.ConvTranspose1d

For usage examples, please refer to the linked torch documentation