`unienv_data.batches.subindex_batch`¶

SubIndexedBatch ¶

SubIndexedBatch(batch: BatchBase[BatchT, BArrayType, BDeviceType, BDtypeType, BRNGType], sub_indexes: Union[IndexableType, BArrayType])

Bases: BatchBase[BatchT, BArrayType, BDeviceType, BDtypeType, BRNGType]

A batch that offers a view of the original batch with a fixed index range. This does not support batch extension.

batch `instance-attribute` ¶

batch = batch

sub_indexes `instance-attribute` ¶

sub_indexes = sub_indexes

is_mutable `property` ¶

is_mutable: bool

backend `property` ¶

backend: ComputeBackend[BArrayType, BDeviceType, BDtypeType, BRNGType]

device `property` ¶

device: Optional[BDeviceType]

single_space `instance-attribute` ¶

single_space = single_space

single_metadata_space `instance-attribute` ¶

single_metadata_space = single_metadata_space

get_flattened_at ¶

get_flattened_at(idx)

get_flattened_at_with_metadata ¶

get_flattened_at_with_metadata(idx)

get_at ¶

get_at(idx)

get_at_with_metadata ¶

get_at_with_metadata(idx) -> Tuple[BatchT, Dict[str, Any]]

set_at ¶

set_at(idx, value)

set_flattened_at ¶

set_flattened_at(idx, value)

extend ¶

extend(value)

extend_flattened ¶

extend_flattened(value)

close ¶

close() -> None

append_flattened ¶

append_flattened(value: BArrayType) -> None

Append one flattened sample to the batch.

remove_at ¶

remove_at(idx: Union[IndexableType, BArrayType]) -> None

Remove one or more samples from the batch.

append ¶

append(value: BatchT) -> None

Append one structured sample to the batch.

extend_from ¶

extend_from(other: BatchBase[BatchT, BArrayType, BDeviceType, BDtypeType, BRNGType], chunk_size: int = 8, tqdm: bool = False) -> None

Copy data from another batch in bounded-size chunks.

get_slice ¶

get_slice(idx: Union[IndexableType, BArrayType]) -> BatchBase[BatchT, BArrayType, BDeviceType, BDtypeType, BRNGType]

Create a lazy view over a subset of indices.

get_column ¶

get_column(nested_keys: Sequence[str]) -> BatchBase[Any, BArrayType, BDeviceType, BDtypeType, BRNGType]

Create a lazy view over a nested field inside each sample.

unienv_data.batches.subindex_batch¶