Convolution modules

class torch_lattice.nn.modules.conv.Conv3d(in_channels, out_channels, kernel_size=3, stride=1, padding=0, dilation=1, bias=False, config=None)

Bases: _BaseConv3d

Support-generating sparse 3D convolution.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• stride (int | Sequence[int])

• padding (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• config (Mapping | None)

class torch_lattice.nn.modules.conv.ConvTranspose3d(in_channels, out_channels, kernel_size=3, stride=1, padding=0, dilation=1, bias=False, config=None)

Bases: _BaseConv3d

Sparse transposed 3D convolution using an existing inverse support map.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• stride (int | Sequence[int])

• padding (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• config (Mapping | None)

class torch_lattice.nn.modules.conv.GenerativeConvTranspose3d(in_channels, out_channels, kernel_size=3, stride=1, padding=0, dilation=1, bias=False, config=None)

Bases: _BaseConv3d

Sparse transposed 3D convolution that generates its output support.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• stride (int | Sequence[int])

• padding (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• config (Mapping | None)

class torch_lattice.nn.modules.conv.NormalizedConvTranspose3d(in_channels, out_channels, kernel_size=3, stride=1, padding=0, dilation=1, bias=False, eps=1e-08, config=None)

Bases: ConvTranspose3d

Weight-normalized sparse transpose convolution.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• stride (int | Sequence[int])

• padding (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• eps (float)

• config (Mapping | None)

class torch_lattice.nn.modules.conv.NormalizedGenerativeConvTranspose3d(in_channels, out_channels, kernel_size=3, stride=1, padding=0, dilation=1, bias=False, eps=1e-08, config=None)

Bases: GenerativeConvTranspose3d

Weight-normalized transpose convolution with generated support.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• stride (int | Sequence[int])

• padding (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• eps (float)

• config (Mapping | None)

class torch_lattice.nn.modules.conv.NormalizedSubmConv3d(in_channels, out_channels, kernel_size=3, dilation=1, bias=False, eps=1e-08, config=None)

Bases: SubmConv3d

Weight-normalized convolution on input coordinate support.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• eps (float)

• config (Mapping | None)

class torch_lattice.nn.modules.conv.SubmConv3d(in_channels, out_channels, kernel_size=3, dilation=1, bias=False, config=None)

Bases: _BaseConv3d

Support-preserving submanifold sparse 3D convolution.

Parameters:

• in_channels (int)

• out_channels (int)

• kernel_size (int | Sequence[int])

• dilation (int | Sequence[int])

• bias (bool)

• config (Mapping | None)