netket.nn.MaskedConv2D#
- class netket.nn.MaskedConv2D[source]#
Bases:
Module
2D convolution module with mask for autoregressive NN.
- Attributes
-
feature_group_count:
int
= 1# 1).
- Type:
if specified, divides the input features into groups (default
-
precision:
Any
= None# numerical precision of the computation, see
jax.lax.Precision
for details.
-
kernel_dilation:
tuple
[int
,int
]# a sequence of 2 integers, giving the dilation factor to apply in each spatial dimension of the convolution kernel.
-
feature_group_count: