Method __init__

scripts/test_random_input.py:15–27

(
		self, in_features, out_features,
		num_hidden_layers, hidden_features,
		activation:str='ReLU', last_activation:str='None',
		seed=42)

class TcnnFCBlock(tcnn.Network):
	def __init__(
		self, in_features, out_features,
		num_hidden_layers, hidden_features,
		activation:str='ReLU', last_activation:str='None',
		seed=42):
		assert hidden_features in [16, 32, 64, 128], "hidden_features can only be 16, 32, 64, or 128."
		super().__init__(in_features, out_features, network_config={
			"otype": "FullyFusedMLP",               # Component type.
			"activation": activation,               # Activation of hidden layers.
			"output_activation": last_activation,   # Activation of the output layer.
			"n_neurons": hidden_features,           # Neurons in each hidden layer. # May only be 16, 32, 64, or 128.
			"n_hidden_layers": num_hidden_layers,   # Number of hidden layers.
		}, seed=seed)

	def forward(self, x: torch.Tensor):
		prefix = x.shape[:-1]