org.bytedeco.pytorch

Class TensorDataset

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.TensorBatchDataset

org.bytedeco.pytorch.TensorDatasetBase

org.bytedeco.pytorch.TensorDataset

All Implemented Interfaces:

AutoCloseable

@Namespace(value="torch::data::datasets")
 @NoOffset
 @Properties(inherit=torch.class)
public class TensorDataset
extends TensorDatasetBase

A dataset of tensors. Stores a single tensor internally, which is then indexed inside get().

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

is_stateful

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
TensorDataset(Pointer p) Pointer cast constructor.
TensorDataset(Tensor tensor)
TensorDataset(TensorVector tensors) Creates a TensorDataset from a vector of tensors.

Method Summary

Modifier and Type	Method and Description
Example	get(long index) Returns a single TensorExample.
SizeTOptional	size() Returns the number of tensors in the dataset.
Tensor	tensor()
TensorDataset	tensor(Tensor setter)

Methods inherited from class org.bytedeco.pytorch.TensorDatasetBase

get_batch, get_batch

Methods inherited from class org.bytedeco.pytorch.TensorBatchDataset

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, trimMemory, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

TensorDataset

public TensorDataset(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

TensorDataset

public TensorDataset(@Const @ByRef
                     TensorVector tensors)

Creates a TensorDataset from a vector of tensors.

TensorDataset

public TensorDataset(@ByVal
                     Tensor tensor)

Method Detail

get

@ByVal
 @Cast(value="torch::data::TensorExample*")
public Example get(@Cast(value="size_t")
                                                                        long index)

Returns a single TensorExample.

Overrides:

get in class TensorDatasetBase

size

@ByVal
public SizeTOptional size()

Returns the number of tensors in the dataset.

Overrides:

size in class TensorBatchDataset

tensor

@ByRef
public Tensor tensor()

tensor

public TensorDataset tensor(Tensor setter)