oneSided.put(forward,Nd4j.scalar(1.0)); INDArray backward = and(upperBound2.lt(lowerBound),not(central)); hAdjusted.put(backward,min(h.get(backward),lowerDist.get(backward).mul(0.5).divi(numSteps))); oneSided.put(backward,Nd4j.scalar(1.0));
@Override public BooleanTensor lessThan(int value) { return fromMask(tensor.lt(value), copyOf(getShape(), getRank())); }
@Override public BooleanTensor lessThan(double value) { return fromMask(tensor.lt(value), copyOf(getShape(), getRank())); }
@Override public BooleanTensor lessThan(DoubleTensor value) { INDArray mask; if (value.isScalar()) { mask = tensor.lt(value.scalar()); } else { INDArray indArray = unsafeGetNd4J(value); mask = INDArrayShim.lt(tensor, indArray); } return fromMask(mask, copyOf(mask.shape(), mask.shape().length)); }
@Override public BooleanTensor lessThan(IntegerTensor value) { INDArray mask; if (value.isScalar()) { mask = tensor.lt(value.scalar()); } else { INDArray indArray = unsafeGetNd4J(value); mask = INDArrayShim.lt(tensor, indArray); } return fromMask(mask, copyOf(mask.shape(), mask.shape().length)); }