netket.nn.FastMaskedConv2D#
- class netket.nn.FastMaskedConv2D[source]#
Bases:
Module
2D convolution module with mask for fast autoregressive NN.
See
netket.models.FastARNNSequential
for a brief explanation of fast autoregressive sampling.- 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_size:
tuple
[int
,int
]# shape of the convolutional kernel (h, w). Typically, \(h = w // 2 + 1\).
-
feature_group_count: