MemSaveConv3d

class memsave_torch.nn.MemSaveConv3d(in_channels: int, out_channels: int, kernel_size, stride=1, padding=0, dilation=1, groups: int = 1, bias: bool = True, padding_mode: str = 'zeros', device=None, dtype=None)

MemSaveConv3d.

Inits a Conv3d layer with the given params.

Parameters:

• in_channels – in_channels

• out_channels – out_channels

• kernel_size – kernel_size

• stride – stride

• padding – padding

• dilation – dilation

• groups – groups

• bias – bias

• padding_mode – padding_mode

• device – device

• dtype – dtype

forward(input: Tensor) → Tensor

Forward pass.

Parameters:

input (torch.Tensor) – Input to the network [B, C_in, D, H, W]

Returns:

Output [B, C_out, D_out, H_out, W_out]

Return type:

torch.Tensor

classmethod from_nn_Conv3d(conv3d: Conv3d)

Converts a nn.Conv3d layer to MemSaveConv3d.

Parameters:

conv3d (nn.Conv3d) – The nn.Conv3d layer

Returns:

The MemSaveConv3d object

Return type:

MemSaveConv3d

Hint

The usage is the same as torch.nn.Conv3d

For usage examples, please refer to the linked torch documentation