Uses of Class
org.nd4j.enums.PartitionMode

Packages that use PartitionMode

Package	Description
org.nd4j.autodiff.samediff.ops
org.nd4j.enums
org.nd4j.linalg.api.ops.impl.shape.tensorops
org.nd4j.linalg.factory.ops

Uses of PartitionMode in org.nd4j.autodiff.samediff.ops

Methods in org.nd4j.autodiff.samediff.ops with parameters of type PartitionMode

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.

Uses of PartitionMode in org.nd4j.enums

Methods in org.nd4j.enums that return PartitionMode

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.

Uses of PartitionMode in org.nd4j.linalg.api.ops.impl.shape.tensorops

Constructors in org.nd4j.linalg.api.ops.impl.shape.tensorops with parameters of type PartitionMode

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)

Uses of PartitionMode in org.nd4j.linalg.factory.ops

Methods in org.nd4j.linalg.factory.ops with parameters of type 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.