Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Index Multiprocessing package - tensorplay.multiprocessing Python Module Index Quantization Search tensorplay tensorplay._stax tensorplay.audio tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision

BatchSampler ChainDataset CheckpointPolicy ConcatDataset DataLoader Dataset GraphExecGroup IterableDataset RandomSampler Sampler SelectiveCheckpointContext SequentialSampler StackDataset Subset SubsetRandomSampler TensorDataset tensorplay.utils.checkpoint.checkpoint tensorplay.utils.checkpoint.checkpoint_sequential tensorplay.utils.checkpoint.create_selective_checkpoint_contexts tensorplay.utils.checkpoint.set_checkpoint_debug_enabled tensorplay.utils.checkpoint.set_checkpoint_early_stop tensorplay.utils.checkpoint.set_device_states tensorplay.utils.data.default_collate tensorplay.utils.data.default_convert tensorplay.utils.data.get_worker_info WeightedRandomSampler

API symbolsutils

latest (dev)

latest (dev)2026-09-08 v1.0.02026-09-06

Copy

SelectiveCheckpointContext

class tensorplay.utils.checkpoint.SelectiveCheckpointContext(*, is_recompute: bool, op_output: Any = None)[source]

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

Sampler

Base class for all Samplers.

SequentialSampler

Samples elements sequentially, always in the same order.

On this page

SelectiveCheckpointContext