| Package | Description |
|---|---|
| org.bytedeco.pytorch | |
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
EmbeddingBagFuncOptions |
EmbeddingBagFuncOptions.getPointer(long i) |
EmbeddingBagFuncOptions |
EmbeddingBagFuncOptions.position(long position) |
| Modifier and Type | Method and Description |
|---|---|
static Tensor |
torch.embedding_bag(Tensor input,
Tensor weight,
EmbeddingBagFuncOptions options)
See
/** https://pytorch.org/docs/master/nn.functional.html#torch.nn.functional.embedding_bag
/** about the exact behavior of this functional.
|
Copyright © 2024. All rights reserved.