![]() |
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
)