@Name(value="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>") @NoOffset @Properties(inherit=torch.class) public class ChunkTensorDataset extends ChunkStatefulTensorDataset
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
is_stateful
Constructor and Description |
---|
ChunkTensorDataset(ChunkTensorDataReader chunk_reader,
RandomSampler chunk_sampler,
RandomSampler example_sampler,
ChunkDatasetOptions options) |
ChunkTensorDataset(ChunkTensorDataReader chunk_reader,
RandomSampler chunk_sampler,
RandomSampler example_sampler,
ChunkDatasetOptions options,
Pointer preprocessing_policy) |
ChunkTensorDataset(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
RandomSampler |
chunk_sampler() |
TensorExampleVectorOptional |
get_batch()
Helper method around get_batch as
batch_size is not strictly necessary |
TensorExampleVectorOptional |
get_batch(long batch_size)
Default get_batch method of BatchDataset.
|
void |
load(InputArchive archive)
Deserializes the statefulDataset's state from the
archive . |
void |
reset()
This will clear any internal state and starts the internal prefetching
mechanism for the chunk dataset.
|
void |
save(OutputArchive archive)
Saves the statefulDataset's state to OutputArchive.
|
SizeTOptional |
size()
size is not used for chunk dataset.
|
is_stateful
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
public ChunkTensorDataset(Pointer p)
Pointer(Pointer)
.public ChunkTensorDataset(ChunkTensorDataReader chunk_reader, RandomSampler chunk_sampler, RandomSampler example_sampler, ChunkDatasetOptions options)
public ChunkTensorDataset(ChunkTensorDataReader chunk_reader, RandomSampler chunk_sampler, RandomSampler example_sampler, ChunkDatasetOptions options, Pointer preprocessing_policy)
@ByVal public TensorExampleVectorOptional get_batch(@Cast(value="size_t") long batch_size)
get_batch
in class ChunkTensorBatchDataset
@ByVal public TensorExampleVectorOptional get_batch()
batch_size
is not strictly necessarypublic void reset()
reset
in class ChunkStatefulTensorDataset
@ByVal public SizeTOptional size()
size
in class ChunkTensorBatchDataset
@Cast(value="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>::ChunkSamplerType*") @ByRef public RandomSampler chunk_sampler()
public void save(@ByRef OutputArchive archive)
ChunkStatefulTensorDataset
save
in class ChunkStatefulTensorDataset
public void load(@ByRef InputArchive archive)
ChunkStatefulTensorDataset
archive
.load
in class ChunkStatefulTensorDataset
Copyright © 2024. All rights reserved.