Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler 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 Upstream anchor labels

Reference guides

Copy

View Markdown Download .md

tensorplay.utils.data

Memory Pinning

`tensorplay.utils.data.DataLoader`	Data loader combines a dataset and a sampler, and provides an iterable over the given dataset.
`tensorplay.utils.data.Dataset`	An abstract class representing a `Dataset`.
`tensorplay.utils.data.IterableDataset`	An iterable Dataset.
`tensorplay.utils.data.TensorDataset`	Dataset wrapping tensors.
`tensorplay.utils.data.StackDataset`	Dataset as a stacking of multiple datasets.
`tensorplay.utils.data.ConcatDataset`	Dataset as a concatenation of multiple datasets.
`tensorplay.utils.data.ChainDataset`	Dataset for chaining multiple `IterableDataset` s.
`tensorplay.utils.data.Subset`	Subset of a dataset at specified indices.
`tensorplay.utils.data.default_collate`	Take in a batch of data and put the elements within the batch into a tensor with an additional outer dimension - batch size.
`tensorplay.utils.data.default_convert`	Convert each NumPy array element into a `tensorplay.Tensor`.
`tensorplay.utils.data.get_worker_info`	Returns the information about the current DataLoader iterator worker process.
`tensorplay.utils.data.Sampler`	Base class for all Samplers.
`tensorplay.utils.data.SequentialSampler`	Samples elements sequentially, always in the same order.
`tensorplay.utils.data.RandomSampler`	Samples elements randomly.
`tensorplay.utils.data.SubsetRandomSampler`	Samples elements randomly from a given list of indices, without replacement.
`tensorplay.utils.data.WeightedRandomSampler`	Samples elements from `[0,..,len(weights)-1]` with given probabilities (weights).
`tensorplay.utils.data.BatchSampler`	Wraps another sampler to yield a mini-batch of indices.

Help improve this page

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

Was this page helpful?

Suggest an edit Report an issue

tensorplay.utils.checkpoint

tensorplay.utils.checkpoint.checkpoint

tensorplay.vision

tensorplay.vision is the computer-vision toolkit of TensorPlay, mirroring the torchvision package layout. The listing below is a static overview; the package requires optional dependencies (Pillow) for some entry points,