Package org.nd4j.linalg.api.ops
Class BaseScalarBoolOp
- java.lang.Object
-
- org.nd4j.autodiff.functions.DifferentialFunction
-
- org.nd4j.linalg.api.ops.BaseOp
-
- org.nd4j.linalg.api.ops.BaseScalarBoolOp
-
- Direct Known Subclasses:
ScalarAnd
,ScalarEps
,ScalarEquals
,ScalarGreaterThan
,ScalarGreaterThanOrEqual
,ScalarLessThan
,ScalarLessThanOrEqual
,ScalarNot
,ScalarNotEquals
,ScalarOr
,ScalarXor
public abstract class BaseScalarBoolOp extends BaseOp implements ScalarOp
-
-
Field Summary
-
Fields inherited from class org.nd4j.linalg.api.ops.BaseOp
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
-
Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
-
-
Constructor Summary
Constructors Constructor Description BaseScalarBoolOp()
BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar)
BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar, boolean inPlace)
BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar, boolean inPlace, Object[] extraArgs)
BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar, Object[] extraArgs)
BaseScalarBoolOp(INDArray x, Number num)
BaseScalarBoolOp(INDArray x, INDArray z, Number set)
BaseScalarBoolOp(INDArray x, INDArray y, INDArray z, Number num)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description List<DataType>
calculateOutputDataTypes(List<DataType> dataTypes)
Calculate the data types for the output arrays.List<LongShapeDescriptor>
calculateOutputShape()
Calculate the output shape for this opList<LongShapeDescriptor>
calculateOutputShape(OpContext oc)
int[]
getDimension()
Op.Type
getOpType()
Op.Type
opType()
The type of the opINDArray
scalar()
The normal scalarvoid
setDimension(int... dimension)
void
setScalar(Number scalar)
This method allows to set scalarvoid
setScalar(INDArray scalar)
boolean
validateDataTypes(boolean experimentalMode)
INDArray
z()
The resulting ndarray-
Methods inherited from class org.nd4j.linalg.api.ops.BaseOp
clearArrays, computeVariables, defineDimensions, dimensions, equals, extraArgs, extraArgsBuff, extraArgsDataBuff, getFinalResult, getInputArgument, getNumOutputs, getOpType, hashCode, initFromOnnx, initFromTensorFlow, onnxName, outputVariables, setX, setY, setZ, tensorflowName, toCustomOp, toString, x, y
-
Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, configureWithSameDiff, diff, doDiff, dup, getBooleanFromProperty, getDoubleValueFromProperty, getIntValueFromProperty, getLongValueFromProperty, getStringFromProperty, getValue, isConfigProperties, larg, mappingsForFunction, onnxNames, opName, opNum, outputs, outputVariable, outputVariables, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface org.nd4j.linalg.api.ops.Op
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, opName, opNum, setExtraArgs, setX, setY, setZ, toCustomOp, x, y
-
Methods inherited from interface org.nd4j.linalg.api.ops.ScalarOp
dimensions
-
-
-
-
Constructor Detail
-
BaseScalarBoolOp
public BaseScalarBoolOp()
-
BaseScalarBoolOp
public BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar)
-
BaseScalarBoolOp
public BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar, boolean inPlace)
-
BaseScalarBoolOp
public BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar, boolean inPlace, Object[] extraArgs)
-
BaseScalarBoolOp
public BaseScalarBoolOp(SameDiff sameDiff, SDVariable i_v, Number scalar, Object[] extraArgs)
-
-
Method Detail
-
calculateOutputShape
public List<LongShapeDescriptor> calculateOutputShape()
Description copied from class:DifferentialFunction
Calculate the output shape for this op- Overrides:
calculateOutputShape
in classDifferentialFunction
- Returns:
- List of output shape descriptors
-
calculateOutputShape
public List<LongShapeDescriptor> calculateOutputShape(OpContext oc)
- Overrides:
calculateOutputShape
in classDifferentialFunction
-
opType
public Op.Type opType()
Description copied from class:DifferentialFunction
The type of the op- Overrides:
opType
in classDifferentialFunction
- Returns:
-
setScalar
public void setScalar(Number scalar)
Description copied from interface:ScalarOp
This method allows to set scalar
-
getDimension
public int[] getDimension()
- Specified by:
getDimension
in interfaceScalarOp
-
setDimension
public void setDimension(int... dimension)
- Specified by:
setDimension
in interfaceScalarOp
-
validateDataTypes
public boolean validateDataTypes(boolean experimentalMode)
- Specified by:
validateDataTypes
in interfaceScalarOp
-
calculateOutputDataTypes
public List<DataType> calculateOutputDataTypes(List<DataType> dataTypes)
Description copied from class:DifferentialFunction
Calculate the data types for the output arrays. Though datatypes can also be inferred fromDifferentialFunction.calculateOutputShape()
, this method differs in that it does not require the input arrays to be populated. This is important as it allows us to do greedy datatype inference for the entire net - even if arrays are not available.- Overrides:
calculateOutputDataTypes
in classDifferentialFunction
- Parameters:
dataTypes
- The data types of the inputs- Returns:
- The data types of the outputs
-
-