torchsig.models.model_utils.simple_models.dense_block

torchsig.models.model_utils.simple_models.dense_block(in_width, out_width, activation_fn=None)[source]

returns a block of layers consisting of a 2d convolution, batch normalization, and activation function, with the input and output channels given by in_width and out_with, a kernel given by kernel_shape, and using the given activation_fn if no activation function is provided, this defaults to ELU