oryx.experimental.nn.convolution.conv_info

Returns parameters and output shape information given input shapes.

oryx.experimental.nn.convolution.conv_info(
    in_shape,
    out_chan,
    filter_shape,
    strides=None,
    padding='VALID',
    kernel_init=None,
    bias_init=stax.randn(1e-06),
    transpose=False
)