Models Module

The models module provides pre-built neural network architectures using logic gate layers.

Pre-built Models

CNN

An implementation of a logic gate convolutional neural network.

Model Details

class torchlogix.models.CNN(class_count, tau, **llkw)[source]

Bases: Module

An implementation of a logic gate convolutional neural network.

__init__(class_count, tau, **llkw)[source]

Initialize internal Module state, shared by both nn.Module and ScriptModule.

forward(x)[source]

Forward pass of the logic gate convolutional neural network.

training: bool