org.bytedeco.pytorch

Class ChunkMapDataset

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.ChunkMapBatchDataset

org.bytedeco.pytorch.ChunkMapDataset

All Implemented Interfaces:

AutoCloseable

@Name(value="torch::data::datasets::MapDataset<torch::data::datasets::SharedBatchDataset<torch::data::datasets::ChunkDataset<JavaCPP_torch_0003a_0003adata_0003a_0003adatasets_0003a_0003aChunkDataReader_0003ctorch_0003a_0003adata_0003a_0003aExample_0003ctorch_0003a_0003aTensor_0002ctorch_0003a_0003aTensor_0003e_0002cstd_0003a_0003avector_0003ctorch_0003a_0003adata_0003a_0003aExample_0003ctorch_0003a_0003aTensor_0002ctorch_0003a_0003aTensor_0003e_00020_0003e_00020_0003e,torch::data::samplers::RandomSampler,torch::data::samplers::RandomSampler> >,torch::data::transforms::Stack<torch::data::Example<torch::Tensor,torch::Tensor> > >")
 @NoOffset
 @Properties(inherit=torch.class)
public class ChunkMapDataset
extends ChunkMapBatchDataset

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.ChunkMapBatchDataset

is_stateful

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
ChunkMapDataset(ChunkSharedBatchDataset dataset, ExampleStack transform)
ChunkMapDataset(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
ChunkSharedBatchDataset	dataset() Returns the underlying dataset.
ExampleOptional	get_batch_example(long indices) Gets a batch from the source dataset and applies the transform to it, returning the result.
SizeTOptional	size() Returns the size of the source dataset.
ExampleStack	transform() Returns the transform being applied.

Methods inherited from class org.bytedeco.pytorch.ChunkMapBatchDataset

get_batch, get_batch, is_stateful

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

ChunkMapDataset

public ChunkMapDataset(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

ChunkMapDataset

public ChunkMapDataset(@ByVal
                       ChunkSharedBatchDataset dataset,
                       @ByVal @Cast(value="torch::data::datasets::MapDataset<torch::data::datasets::SharedBatchDataset<torch::data::datasets::ChunkDataset<JavaCPP_torch_0003a_0003adata_0003a_0003adatasets_0003a_0003aChunkDataReader_0003ctorch_0003a_0003adata_0003a_0003aExample_0003ctorch_0003a_0003aTensor_0002ctorch_0003a_0003aTensor_0003e_0002cstd_0003a_0003avector_0003ctorch_0003a_0003adata_0003a_0003aExample_0003ctorch_0003a_0003aTensor_0002ctorch_0003a_0003aTensor_0003e_00020_0003e_00020_0003e,torch::data::samplers::RandomSampler,torch::data::samplers::RandomSampler> >,torch::data::transforms::Stack<torch::data::Example<torch::Tensor,torch::Tensor> > >::TransformType*")
                       ExampleStack transform)

Method Detail

get_batch_example

@Name(value="get_batch")
 @ByVal
public ExampleOptional get_batch_example(@Cast(value="size_t")
                                                                          long indices)

Gets a batch from the source dataset and applies the transform to it, returning the result.

size

@ByVal
 @NoException(value=true)
public SizeTOptional size()

Returns the size of the source dataset.

Overrides:

size in class ChunkMapBatchDataset

dataset

@Const
 @ByRef
 @NoException(value=true)
public ChunkSharedBatchDataset dataset()

Returns the underlying dataset.

transform

@Const
 @ByRef
 @NoException(value=true)
public ExampleStack transform()

Returns the transform being applied.