quantax.nn.ReshapeConv

class quantax.nn.ReshapeConv

Bases: Module

Reshape the input to the shape suitable for convolutional layers.

A fock state in Quantax is usually givne by a 1D array with entries +1/-1. This layer reshape it to shape.

__init__(dtype: ~numpy.dtype = <class 'jax.numpy.float32'>)

Parameters:

dtype – Convert the input to the given data type, by default float32.

Attributes

dtype