@Name(value="torch::data::datasets::Dataset<torch::data::datasets::TensorDataset,torch::data::TensorExample>") @Properties(inherit=torch.class) public class TensorDatasetBase extends TensorBatchDataset
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
is_stateful
Constructor and Description |
---|
TensorDatasetBase(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
TensorExampleVector |
get_batch(long... request) |
TensorExampleVector |
get_batch(SizeTArrayRef request)
Returns a batch of data.
|
Example |
get(long index)
Returns the example at the given index.
|
is_stateful, size
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 TensorDatasetBase(Pointer p)
Pointer(Pointer)
.@ByVal @Cast(value="torch::data::datasets::Dataset<torch::data::datasets::TensorDataset,torch::data::TensorExample>::ExampleType*") public Example get(@Cast(value="size_t") long index)
@ByVal public TensorExampleVector get_batch(@ByVal SizeTArrayRef request)
get()
for every requested index
in the batch.get_batch
in class TensorBatchDataset
@ByVal public TensorExampleVector get_batch(@ByVal @Cast(value={"size_t*","c10::ArrayRef<size_t>","std::vector<size_t>&"}) @StdVector(value="size_t") long... request)
get_batch
in class TensorBatchDataset
Copyright © 2024. All rights reserved.