@Namespace(value="c10d") @NoOffset @Properties(inherit=torch.class) public class GradBucket extends Pointer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
GradBucket(long index,
long bucket_count,
Tensor tensor,
SizeTVector offsets,
SizeTVector lengths,
LongArrayRefVector sizes_vec,
TensorVector parameters,
TensorOptional sparse_grad_indices) |
GradBucket(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
Tensor |
getBuffer() |
Tensor |
getBufferRef() |
TensorVector |
getGradients() |
long |
getIndex() |
TensorVector |
getParameters() |
TensorOptional |
getSparseGradIndices() |
boolean |
isLast() |
void |
setBuffer(Tensor buffer) |
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 GradBucket(Pointer p)
Pointer(Pointer)
.public GradBucket(@Cast(value="size_t") long index, @Cast(value="size_t") long bucket_count, @ByVal Tensor tensor, @ByVal @Cast(value="std::vector<size_t>*") SizeTVector offsets, @ByVal @Cast(value="std::vector<size_t>*") SizeTVector lengths, @ByVal LongArrayRefVector sizes_vec, @ByVal TensorVector parameters, @ByVal TensorOptional sparse_grad_indices)
@ByVal public TensorVector getGradients()
@Const @ByVal public TensorVector getParameters()
@ByRef public TensorOptional getSparseGradIndices()
Copyright © 2024. All rights reserved.