netket.nn.MaskedConv1D#
- class netket.nn.MaskedConv1D[source]#
Bases:
Module
1D 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.
-
feature_group_count: