tabensemb.model.AbstractNN.example_input_array

property

property AbstractNN.example_input_array: Tensor | Tuple | Dict | None

The example input array is a specification of what the module can consume in the forward() method. The return type is interpreted as follows:

• Single tensor: It is assumed the model takes a single argument, i.e., model.forward(model.example_input_array)

• Tuple: The input array should be interpreted as a sequence of positional arguments, i.e., model.forward(*model.example_input_array)

• Dict: The input array represents named keyword arguments, i.e., model.forward(**model.example_input_array)