Package | Description |
---|---|
org.nd4j.autodiff.samediff.ops | |
org.nd4j.enums | |
org.nd4j.linalg.api.ops.impl.shape.tensorops | |
org.nd4j.linalg.factory.ops |
Modifier and Type | Method and Description |
---|---|
SDVariable |
SDMath.embeddingLookup(SDVariable x,
SDVariable indices,
PartitionMode PartitionMode)
Looks up ids in a list of embedding tensors.
|
SDVariable |
SDMath.embeddingLookup(String name,
SDVariable x,
SDVariable indices,
PartitionMode PartitionMode)
Looks up ids in a list of embedding tensors.
|
Modifier and Type | Method and Description |
---|---|
static PartitionMode |
PartitionMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static PartitionMode[] |
PartitionMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Constructor and Description |
---|
EmbeddingLookup(@NonNull INDArray in,
@NonNull INDArray indices,
PartitionMode partitionMode,
INDArray output) |
EmbeddingLookup(@NonNull INDArray in,
INDArray output,
PartitionMode partitionMode,
int... indices) |
EmbeddingLookup(@NonNull SameDiff sameDiff,
@NonNull SDVariable in,
@NonNull SDVariable indices,
PartitionMode partitionMode) |
Modifier and Type | Method and Description |
---|---|
INDArray |
NDMath.embeddingLookup(INDArray x,
INDArray indices,
PartitionMode PartitionMode)
Looks up ids in a list of embedding tensors.
|
Copyright © 2020. All rights reserved.