PFNN

class pinnx.nn.PFNN(layer_sizes, activation, kernel_initializer=KaimingUniform(mode=fan_in, nonlinearity=relu, unit=1), input_transform=None, output_transform=None)

Parallel fully-connected network that uses independent sub-networks for each network output.

Parameters:

layer_sizes (Sequence[int]) – A nested list that defines the architecture of the neural network (how the layers are connected). If layer_sizes[i] is an int, it represents one layer shared by all the outputs; if layer_sizes[i] is a list, it represents len(layer_sizes[i]) sub-layers, each of which is exclusively used by one output. Note that len(layer_sizes[i]) should equal the number of outputs. Every number specifies the number of neurons in that layer.