org.bytedeco.pytorch

Class ChunkMapTensorDataset

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.ChunkMapTensorBatchDataset

org.bytedeco.pytorch.ChunkMapTensorDataset

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_0003adata_0003a_0003aexample_0003a_0003aNoTarget_0003e_0002cstd_0003a_0003avector_0003ctorch_0003a_0003adata_0003a_0003aExample_0003ctorch_0003a_0003aTensor_0002ctorch_0003a_0003adata_0003a_0003aexample_0003a_0003aNoTarget_0003e_00020_0003e_00020_0003e,torch::data::samplers::RandomSampler,torch::data::samplers::RandomSampler> >,torch::data::transforms::Stack<torch::data::Example<torch::Tensor,torch::data::example::NoTarget> > >")
 @NoOffset
 @Properties(inherit=torch.class)
public class ChunkMapTensorDataset
extends ChunkMapTensorBatchDataset

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

is_stateful

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
ChunkMapTensorDataset(ChunkSharedTensorBatchDataset dataset, TensorExampleStack transform)
ChunkMapTensorDataset(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
ChunkSharedTensorBatchDataset	dataset() Returns the underlying dataset.
TensorExampleOptional	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.
TensorExampleStack	transform() Returns the transform being applied.

Methods inherited from class org.bytedeco.pytorch.ChunkMapTensorBatchDataset

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

ChunkMapTensorDataset

public ChunkMapTensorDataset(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

ChunkMapTensorDataset

public ChunkMapTensorDataset(@ByVal
                             ChunkSharedTensorBatchDataset 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_0003adata_0003a_0003aexample_0003a_0003aNoTarget_0003e_0002cstd_0003a_0003avector_0003ctorch_0003a_0003adata_0003a_0003aExample_0003ctorch_0003a_0003aTensor_0002ctorch_0003a_0003adata_0003a_0003aexample_0003a_0003aNoTarget_0003e_00020_0003e_00020_0003e,torch::data::samplers::RandomSampler,torch::data::samplers::RandomSampler> >,torch::data::transforms::Stack<torch::data::Example<torch::Tensor,torch::data::example::NoTarget> > >::TransformType*")
                             TensorExampleStack transform)

Method Detail

get_batch_example

@Name(value="get_batch")
 @ByVal
public TensorExampleOptional 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 ChunkMapTensorBatchDataset

dataset

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

Returns the underlying dataset.

transform

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

Returns the transform being applied.