.. currentmodule:: symjax.nn.layers
.. autoclass:: Conv2D
.. automethod:: __init__
.. rubric:: Methods
.. autosummary::
~Conv2D.__init__
~Conv2D.add_update
~Conv2D.add_variable
~Conv2D.argmax
~Conv2D.argmin
~Conv2D.astype
~Conv2D.cast
~Conv2D.clone
~Conv2D.conj
~Conv2D.conjugate
~Conv2D.create_tensor
~Conv2D.create_variable
~Conv2D.dot
~Conv2D.expand_dims
~Conv2D.flatten
~Conv2D.forward
~Conv2D.imag
~Conv2D.init_input
~Conv2D.matmul
~Conv2D.max
~Conv2D.mean
~Conv2D.min
~Conv2D.prod
~Conv2D.real
~Conv2D.repeat
~Conv2D.reshape
~Conv2D.round
~Conv2D.squeeze
~Conv2D.std
~Conv2D.sum
~Conv2D.transpose
~Conv2D.var
~Conv2D.variables
.. rubric:: Attributes
.. autosummary::
~Conv2D.dtype
~Conv2D.fn_name
~Conv2D.name
~Conv2D.ndim
~Conv2D.scope
~Conv2D.shape
~Conv2D.updates