@Namespace(value="torch::nn") @NoOffset @Properties(inherit=torch.class) public class EmbeddingImpl extends EmbeddingImplCloneable
torch::nn::EmbeddingOptions class to learn what
constructor arguments are supported for this module.
Example:
Embedding model(EmbeddingOptions(10,
2).padding_idx(3).max_norm(2).norm_type(2.5).scale_grad_by_freq(true).sparse(true));
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter| Constructor and Description |
|---|
EmbeddingImpl(EmbeddingOptions options_) |
EmbeddingImpl(long num_embeddings,
long embedding_dim) |
EmbeddingImpl(Pointer p)
Pointer cast constructor.
|
| Modifier and Type | Method and Description |
|---|---|
Tensor |
forward(Tensor indices)
Performs a lookup on the embedding table stored in
weight using the
indices supplied and returns the result. |
EmbeddingOptions |
options()
The
Options used to configure this Embedding module. |
EmbeddingImpl |
options(EmbeddingOptions setter) |
void |
pretty_print(Pointer stream)
Pretty prints the
Embedding module into the given stream. |
void |
reset_parameters() |
void |
reset()
reset() must perform initialization of all members with reference
semantics, most importantly parameters, buffers and submodules. |
Tensor |
weight()
The embedding table.
|
EmbeddingImpl |
weight(Tensor setter) |
clone, cloneapply, apply, apply, apply, apply, apply, apply, apply, asAdaptiveAvgPool1d, asAdaptiveAvgPool2d, asAdaptiveAvgPool3d, asAdaptiveLogSoftmaxWithLoss, asAdaptiveMaxPool1d, asAdaptiveMaxPool2d, asAdaptiveMaxPool3d, asAlphaDropout, asAvgPool1d, asAvgPool2d, asAvgPool3d, asBatchNorm1d, asBatchNorm2d, asBatchNorm3d, asBCELoss, asBCEWithLogitsLoss, asBilinear, asCELU, asConstantPad1d, asConstantPad2d, asConstantPad3d, asConv1d, asConv2d, asConv3d, asConvTranspose1d, asConvTranspose2d, asConvTranspose3d, asCosineEmbeddingLoss, asCosineSimilarity, asCrossEntropyLoss, asCrossMapLRN2d, asCTCLoss, asDropout, asDropout2d, asDropout3d, asELU, asEmbedding, asEmbeddingBag, asFeatureAlphaDropout, asFlatten, asFold, asFractionalMaxPool2d, asFractionalMaxPool3d, asGELU, asGLU, asGroupNorm, asGRU, asGRUCell, asHardshrink, asHardtanh, asHingeEmbeddingLoss, asHuberLoss, asIdentity, asInstanceNorm1d, asInstanceNorm2d, asInstanceNorm3d, asKLDivLoss, asL1Loss, asLayerNorm, asLeakyReLU, asLinear, asLocalResponseNorm, asLogSigmoid, asLogSoftmax, asLPPool1d, asLPPool2d, asLPPool3d, asLSTM, asLSTMCell, asMarginRankingLoss, asMaxPool1d, asMaxPool2d, asMaxPool3d, asMaxUnpool1d, asMaxUnpool2d, asMaxUnpool3d, asMish, asModuleDict, asModuleList, asMSELoss, asMultiheadAttention, asMultiLabelMarginLoss, asMultiLabelSoftMarginLoss, asMultiMarginLoss, asNLLLoss, asPairwiseDistance, asParameterDict, asParameterList, asPixelShuffle, asPixelUnshuffle, asPoissonNLLLoss, asPReLU, asReflectionPad1d, asReflectionPad2d, asReflectionPad3d, asReLU, asReLU6, asReplicationPad1d, asReplicationPad2d, asReplicationPad3d, asRNN, asRNNCell, asRReLU, asSELU, asSequential, asSigmoid, asSiLU, asSmoothL1Loss, asSoftMarginLoss, asSoftmax, asSoftmax2d, asSoftmin, asSoftplus, asSoftshrink, asSoftsign, asTanh, asTanhshrink, asThreshold, asTransformer, asTransformerDecoder, asTransformerDecoderLayer, asTransformerEncoder, asTransformerEncoderLayer, asTripletMarginLoss, asTripletMarginWithDistanceLoss, asUnflatten, asUnfold, asUpsample, asZeroPad1d, asZeroPad2d, asZeroPad3d, buffers, buffers, children, eval, is_serializable, is_training, load, modules, modules, name, named_buffers, named_buffers, named_children, named_modules, named_modules, named_modules, named_parameters, named_parameters, parameters, parameters, put, register_buffer, register_buffer, register_module, register_module, register_parameter, register_parameter, register_parameter, register_parameter, save, shiftLeft, to, to, to, train, unregister_module, unregister_module, zero_gradaddress, 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, zeropublic EmbeddingImpl(Pointer p)
Pointer(Pointer).public EmbeddingImpl(@Cast(value="int64_t") long num_embeddings, @Cast(value="int64_t") long embedding_dim)
public EmbeddingImpl(@ByVal EmbeddingOptions options_)
public void reset()
EmbeddingImplCloneablereset() must perform initialization of all members with reference
semantics, most importantly parameters, buffers and submodules.reset in class EmbeddingImplCloneablepublic void reset_parameters()
public void pretty_print(@Cast(value="std::ostream*") @ByRef Pointer stream)
Embedding module into the given stream.pretty_print in class Module@ByVal public Tensor forward(@Const @ByRef Tensor indices)
weight using the
indices supplied and returns the result.@ByRef public EmbeddingOptions options()
Options used to configure this Embedding module.
Changes to EmbeddingOptions *after construction* have no effect.public EmbeddingImpl options(EmbeddingOptions setter)
public EmbeddingImpl weight(Tensor setter)
Copyright © 2024. All rights reserved.