rl_equation_solver.utilities.utilities.Batch

class Batch[source]

Bases: object

Graph Embedding or state vector Batch

Initialize the batch

Methods

classmethod __call__(states, device)[source]: Batch states for given set of states and send to device. States can be either instances of GraphEmbedding or np.ndarray