NN

class pinnx.nn.NN(input_transform=None, output_transform=None)

Base class for all neural network modules.

apply_feature_transform(transform)

Compute the features by applying a transform to the network inputs, i.e., features = transform(inputs). Then, outputs = network(features).

apply_output_transform(transform)

Apply a transform to the network outputs, i.e., outputs = transform(inputs, outputs).

num_trainable_parameters()

Evaluate the number of trainable parameters for the NN.