cppn-abstract-pyxll-funcs.py

from pyxll import xl_func
import torch.nn as nn


@xl_func
def nn_Linear(in_features: int, out_features: int, bias: bool=True):
    """Create a linear transformation layer."""
    return nn.Linear(in_features, out_features, bias)


@xl_func
def nn_Tanh():
    """Create a Tanh transformation layer."""
    return nn.Tanh()


@xl_func
def nn_Sigmoid():
    """Create a Sigmoid transformation layer."""
    return nn.Sigmoid()