gensbi.recipes.utils#

Functions#

init_ids_1d(dim[, semantic_id])

init_ids_2d(dim[, semantic_id])

init_ids_joint(dim_obs, dim_cond)

patchify_2d(x)

Module Contents#

gensbi.recipes.utils.init_ids_1d(dim, semantic_id=None)[source]#
Parameters:

  • dim (int)

  • semantic_id (Union[int, None])

gensbi.recipes.utils.init_ids_2d(dim, semantic_id=0)[source]#
Parameters:

  • dim (Tuple[int, int])

  • semantic_id (int)

gensbi.recipes.utils.init_ids_joint(dim_obs, dim_cond)[source]#
Parameters:

  • dim_obs (int)

  • dim_cond (int)

gensbi.recipes.utils.patchify_2d(x)[source]#
Parameters:

x (jax.Array)