Function conv1x1

src/shape_assembly/models/encoder/vn_layers.py:12–20

(in_channels, out_channels, dim)

EPS = 1e-6

def conv1x1(in_channels, out_channels, dim):
    if dim == 3:
        return nn.Conv1d(in_channels, out_channels, 1, bias=False)
    elif dim == 4:
        return nn.Conv2d(in_channels, out_channels, 1, bias=False)
    elif dim == 5:
        return nn.Conv3d(in_channels, out_channels, 1, bias=False)
    else:
        raise NotImplementedError(f'{dim}D 1x1 Conv is not supported')


class VNLinear(nn.Module):