org.bytedeco.pytorch

Class MNISTDataset

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.MNISTBatchDataset

org.bytedeco.pytorch.MNISTDataset

All Implemented Interfaces:

AutoCloseable

Direct Known Subclasses:

MNIST

@Name(value="torch::data::datasets::Dataset<torch::data::datasets::MNIST,torch::data::Example<torch::Tensor,torch::Tensor> >")
 @Properties(inherit=torch.class)
public class MNISTDataset
extends MNISTBatchDataset

A dataset that can yield data in batches, or as individual examples. A Dataset is a BatchDataset, because it supports random access and therefore batched access is implemented (by default) by calling the random access indexing function for each index in the requested batch of indices. This can be customized.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.pytorch.MNISTBatchDataset

is_stateful

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
MNISTDataset(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
ExampleVector	get_batch(long... indices)
ExampleVector	get_batch(SizeTArrayRef indices) Returns a batch of data.
Example	get(long index) Returns the example at the given index.

Methods inherited from class org.bytedeco.pytorch.MNISTBatchDataset

is_stateful, map, size

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

MNISTDataset

public MNISTDataset(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

get

@ByVal
 @Cast(value="torch::data::datasets::Dataset<torch::data::datasets::MNIST,torch::data::Example<torch::Tensor,torch::Tensor> >::ExampleType*")
public Example get(@Cast(value="size_t")
                                                                                                                                                                          long index)

Returns the example at the given index.

get_batch

@ByVal
public ExampleVector get_batch(@ByVal
                                       SizeTArrayRef indices)

Returns a batch of data. The default implementation calls get() for every requested index in the batch.

Overrides:

get_batch in class MNISTBatchDataset

get_batch

@ByVal
public ExampleVector get_batch(@ByVal @Cast(value={"size_t*","c10::ArrayRef<size_t>","std::vector<size_t>&"}) @StdVector(value="size_t")
                                       long... indices)

Overrides:

get_batch in class MNISTBatchDataset