org.bytedeco.pytorch

Class MNISTBatchDataset

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.MNISTBatchDataset

All Implemented Interfaces:

AutoCloseable

Direct Known Subclasses:

MNISTDataset

@Name(value="torch::data::datasets::BatchDataset<torch::data::datasets::MNIST,std::vector<torch::data::Example<torch::Tensor,torch::Tensor> >,at::ArrayRef<size_t> >")
 @Properties(inherit=torch.class)
public class MNISTBatchDataset
extends Pointer

A dataset that can yield data only in batches.

Nested Class Summary

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

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

Field Summary

Fields

Modifier and Type	Field and Description
static boolean	is_stateful

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
ExampleVector	get_batch(long... request)
ExampleVector	get_batch(SizeTArrayRef request) Returns a batch of data given an index.
static boolean	is_stateful()
MNISTMapDataset	map(ExampleStack transform) Creates a MapDataset that applies the given transform to this dataset.
SizeTOptional	size() Returns the size of the dataset, or an empty std::optional if it is unsized.

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

Field Detail

is_stateful

public static final boolean is_stateful

Constructor Detail

MNISTBatchDataset

public MNISTBatchDataset(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

is_stateful

@MemberGetter
 @Cast(value="const bool")
public static boolean is_stateful()

get_batch

@ByVal
public ExampleVector get_batch(@ByVal
                                      SizeTArrayRef request)

Returns a batch of data given an index.

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... request)

size

@ByVal
public SizeTOptional size()

Returns the size of the dataset, or an empty std::optional if it is unsized.

map

@ByVal
public MNISTMapDataset map(@ByVal
                                  ExampleStack transform)

Creates a MapDataset that applies the given transform to this dataset.