org.bytedeco.pytorch

Class DistributedRandomSampler

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Sampler

org.bytedeco.pytorch.DistributedSampler

org.bytedeco.pytorch.DistributedRandomSampler

All Implemented Interfaces:

AutoCloseable

@Namespace(value="torch::data::samplers")
 @NoOffset
 @Properties(inherit=torch.class)
public class DistributedRandomSampler
extends DistributedSampler

Select samples randomly. The sampling order is shuffled at each reset() call.

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.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
DistributedRandomSampler(long size)
DistributedRandomSampler(long size, long num_replicas, long rank, boolean allow_duplicates)
DistributedRandomSampler(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
long	index() Returns the current index of the DistributedRandomSampler.
void	load(InputArchive archive) Deserializes the DistributedRandomSampler from the archive.
SizeTVectorOptional	next(long batch_size) Returns the next batch of indices.
void	reset()
void	reset(SizeTOptional new_size) Resets the DistributedRandomSampler to a new set of indices.
void	save(OutputArchive archive) Serializes the DistributedRandomSampler to the archive.

Methods inherited from class org.bytedeco.pytorch.DistributedSampler

epoch, set_epoch

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

DistributedRandomSampler

public DistributedRandomSampler(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

DistributedRandomSampler

public DistributedRandomSampler(@Cast(value="size_t")
                                long size,
                                @Cast(value="size_t")
                                long num_replicas,
                                @Cast(value="size_t")
                                long rank,
                                @Cast(value="bool")
                                boolean allow_duplicates)

DistributedRandomSampler

public DistributedRandomSampler(@Cast(value="size_t")
                                long size)

Method Detail

reset

public void reset(@ByVal(nullValue="c10::optional<size_t>(c10::nullopt)")
                  SizeTOptional new_size)

Resets the DistributedRandomSampler to a new set of indices.

Overrides:

reset in class Sampler

reset

public void reset()

next

@ByVal
public SizeTVectorOptional next(@Cast(value="size_t")
                                       long batch_size)

Returns the next batch of indices.

Overrides:

next in class Sampler

save

public void save(@ByRef
                 OutputArchive archive)

Serializes the DistributedRandomSampler to the archive.

Overrides:

save in class Sampler

load

public void load(@ByRef
                 InputArchive archive)

Deserializes the DistributedRandomSampler from the archive.

Overrides:

load in class Sampler

index

@Cast(value="size_t")
 @NoException(value=true)
public long index()

Returns the current index of the DistributedRandomSampler.