cppn-abstract-create-network.py

from pyxll import xl_func
import torch.nn as nn


class NN(nn.Module):
    """Neural networks with Sequential layers"""

    def __init__(self, layers):
        super(NN, self).__init__()
        self.layers = nn.Sequential(*layers)

    def forward(self, x):
        return self.layers(x)


@xl_func("object[]: object")
def nn_Sequential(layers):
    """Create a neural network from a list of layers."""
    return NN(layers)