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