View source on GitHub |
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
)