/** * * @param relStep * @param x * @return */ public static INDArray computeAbsoluteStep(INDArray relStep,INDArray x) { if(relStep == null) { relStep = pow(Nd4j.scalar(getEpsRelativeTo(x)),0.5); } INDArray signX0 = x.gte(0).muli(2).subi(1); return signX0.mul(relStep).muli(max(abs(x),1.0)); }
@Override public BooleanTensor greaterThanOrEqual(double value) { return fromMask(tensor.gte(value), copyOf(getShape(), getRank())); }
@Override public BooleanTensor greaterThanOrEqual(int value) { return fromMask(tensor.gte(value), copyOf(getShape(), getRank())); }
@Override public BooleanTensor greaterThanOrEqual(IntegerTensor value) { INDArray mask; if (value.isScalar()) { mask = tensor.gte(value.scalar()); } else { INDArray indArray = unsafeGetNd4J(value); mask = tensor.dup(); mask = INDArrayShim.getGreaterThanOrEqualToMask(mask, indArray, BUFFER_TYPE); } return fromMask(mask, copyOf(mask.shape(), mask.shape().length)); }
@Override public BooleanTensor greaterThanOrEqual(DoubleTensor value) { INDArray mask; if (value.isScalar()) { mask = tensor.gte(value.scalar()); } else { INDArray indArray = unsafeGetNd4J(value); mask = tensor.dup(); mask = INDArrayShim.getGreaterThanOrEqualToMask(mask, indArray, BUFFER_TYPE); } return fromMask(mask, copyOf(mask.shape(), mask.shape().length)); }
INDArray geqBinLower = p.gte(j * binSize); INDArray ltBinUpper; if (j == reliabilityDiagNumBins - 1) { INDArray geqBinLower = labelsSubPredicted.gte(j * binSize); INDArray ltBinUpper; INDArray geqBinLowerProbs = maskedProbs.gte(j * binSize); INDArray ltBinUpperProbs; if (j == histogramNumBins - 1) {