Package | Description |
---|---|
org.bytedeco.pytorch | |
org.bytedeco.pytorch.global |
Modifier and Type | Method and Description |
---|---|
EmbeddingBagMode |
EmbeddingBagFromPretrainedOptions.mode() |
EmbeddingBagMode |
EmbeddingBagFuncOptions.mode() |
EmbeddingBagMode |
EmbeddingBagOptions.mode() |
EmbeddingBagMode |
EmbeddingBagMode.put(EmbeddingBagMode x) |
EmbeddingBagMode |
EmbeddingBagMode.put(kMax value) |
EmbeddingBagMode |
EmbeddingBagMode.put(kMean value) |
EmbeddingBagMode |
EmbeddingBagMode.put(kSum value) |
Modifier and Type | Method and Description |
---|---|
static kSum |
EmbeddingBagMode.get0(EmbeddingBagMode container) |
static kMean |
EmbeddingBagMode.get1(EmbeddingBagMode container) |
static kMax |
EmbeddingBagMode.get2(EmbeddingBagMode container) |
EmbeddingBagMode |
EmbeddingBagMode.put(EmbeddingBagMode x) |
Modifier and Type | Method and Description |
---|---|
static Tensor |
torch.embedding_bag(Tensor input,
Tensor weight,
Tensor offsets,
DoubleOptional max_norm,
double norm_type,
boolean scale_grad_by_freq,
EmbeddingBagMode mode,
boolean sparse,
Tensor per_sample_weights,
boolean include_last_offset,
LongOptional padding_idx) |
Copyright © 2024. All rights reserved.