utils

Functions

`flatten`
`get_device`
`set_device`
`speye`	identity matrix of dimension n as sparse_coo_tensor.

Classes

`EarlyStopping`	Context manager for early stopping
`Timer`	Context manager for accumulating execution time