torchsig.models.model_utils.simple_models.double_image_scale_2d¶
- torchsig.models.model_utils.simple_models.double_image_scale_2d(width=1, kernel_shape=[5, 5], activation_fn=None)[source]¶
doubles the scale dimensions of an image of channel width width using a transposd convolution of kernel shape kernel_shape calls batch norm and an activation function provided by activation_fn on the result if no function is provided, this defaults to ELU