long[] retShape = Shape.wholeArrayDimension(dimension) ? new long[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
protected void buildZ(IndexAccumulation op, int... dimension) { Arrays.sort(dimension); for (int i = 0; i < dimension.length; i++) { if (dimension[i] < 0) dimension[i] += op.x().rank(); } //do op along all dimensions if (dimension.length == op.x().rank()) dimension = new int[] {Integer.MAX_VALUE}; int[] retShape = Shape.wholeArrayDimension(dimension) ? new int[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension); //ensure vector is proper shape if (retShape.length == 1) { if (dimension[0] == 0) retShape = new int[] {1, retShape[0]}; else retShape = new int[] {retShape[0], 1}; } else if (retShape.length == 0) { retShape = new int[] {1, 1}; } INDArray ret = null; if (Math.abs(op.zeroDouble()) < Nd4j.EPS_THRESHOLD) { ret = Nd4j.zeros(retShape); } else { ret = Nd4j.valueArrayOf(retShape, op.zeroDouble()); } op.setZ(ret); }
long[] retShape = Shape.wholeArrayDimension(dimension) ? new long[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
if (Shape.wholeArrayDimension(dimension)) retShape = new long[] {1, 1}; else
int[] retShape = Shape.wholeArrayDimension(dimension) ? new int[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
int[] retShape = Shape.wholeArrayDimension(dimension) ? new int[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
dimension = new int[] {Integer.MAX_VALUE}; int[] retShape = Shape.wholeArrayDimension(dimension) ? new int[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
dimension = new int[] {Integer.MAX_VALUE}; long[] retShape = Shape.wholeArrayDimension(dimension) ? new long[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
? AtomicAllocator.getInstance().getPointer(op.extraArgsDataBuff(), context) : null; long[] retShape = Shape.wholeArrayDimension(dimension) ? new long[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
? AtomicAllocator.getInstance().getPointer(op.extraArgsDataBuff(), context) : null; int[] retShape = Shape.wholeArrayDimension(dimension) ? new int[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
dimension = new int[] {Integer.MAX_VALUE}; int[] retShape = Shape.wholeArrayDimension(dimension) ? new int[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);
dimension = new int[] {Integer.MAX_VALUE}; long[] retShape = Shape.wholeArrayDimension(dimension) ? new long[] {1, 1} : ArrayUtil.removeIndex(op.x().shape(), dimension);