org.nd4j.linalg.api.ops.random.impl

Class Choice

java.lang.Object

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.random.BaseRandomOp

org.nd4j.linalg.api.ops.random.impl.Choice

All Implemented Interfaces:

Op, RandomOp

public class Choice
extends BaseRandomOp

This Op implements numpy.choice method It fills Z from source, following probabilities for each source element

Author:

raver119@gmail.com

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.linalg.api.ops.Op

Op.Type

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.BaseOp

extraArgs, extraArgz, n, numProcessed, passThrough, x, y, z

Constructor Summary

Constructors

Constructor and Description
Choice()
Choice(INDArray source, INDArray probabilities, INDArray z)

Method Summary

Modifier and Type	Method and Description
boolean	isExecSpecial() Whether the executioner needs to do a special call or not
String	name() The name of this operation
int	opNum() An op number

Methods inherited from class org.nd4j.linalg.api.ops.random.BaseRandomOp

op, op, op, op, op, op, op, op, opForDimension, opForDimension

Methods inherited from class org.nd4j.linalg.api.ops.BaseOp

exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, getOpType, init, isPassThrough, n, numProcessed, setN, setX, setY, setZ, toString, x, y, z

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.nd4j.linalg.api.ops.Op

exec, exec, extraArgs, extraArgsBuff, extraArgsDataBuff, init, isPassThrough, n, numProcessed, setExtraArgs, setN, setX, setY, setZ, x, y, z

Constructor Detail

Choice

public Choice()

Choice

public Choice(@NonNull
              INDArray source,
              @NonNull
              INDArray probabilities,
              @NonNull
              INDArray z)

Method Detail

isExecSpecial

public boolean isExecSpecial()

Description copied from interface: Op

Whether the executioner needs to do a special call or not

Specified by:

isExecSpecial in interface Op

Overrides:

isExecSpecial in class BaseOp

Returns:

true if the executioner needs to do a special call or not false otherwise

opNum

public int opNum()

Description copied from interface: Op

An op number

Returns:

name

public String name()

Description copied from interface: Op

The name of this operation

Returns:

the name of this operation