@Override public INDArray amean(int... dimension) { return Nd4j.getExecutioner().exec(new AMean(this), dimension); }
@Override public String tensorflowName() { throw new NoOpNameFoundException("No tensorflow op opName found for " + opName()); }
@Override public Op opForDimension(int index, int dimension) { INDArray xAlongDimension = x.vectorAlongDimension(index, dimension); AMean ret; if (y() != null) ret = new AMean(xAlongDimension, y.vectorAlongDimension(index, dimension), xAlongDimension.length()); else ret = new AMean(x.vectorAlongDimension(index, dimension)); ret.setApplyFinalTransform(applyFinalTransform()); return ret; }
@Override public double getAndSetFinalResult(double accum) { double result; if (applyFinalTransform()) { result = accum / n(); this.finalResult = result; } else { result = accum; this.finalResult = result; } return result; }
@Override public double calculateFinalResult(double accum, long n) { if (applyFinalTransform()) return accum / n; return accum; }
@Override public IComplexNumber getAndSetFinalResult(IComplexNumber accum) { finalResultComplex = accum.div(n()); return finalResultComplex; } }
@Override public Op opForDimension(int index, int... dimension) { INDArray xAlongDimension = x.tensorAlongDimension(index, dimension); AMean ret; if (y() != null) ret = new AMean(xAlongDimension, y.tensorAlongDimension(index, dimension), xAlongDimension.length()); else ret = new AMean(x.tensorAlongDimension(index, dimension)); ret.setApplyFinalTransform(applyFinalTransform()); return ret; }
@Override public float getAndSetFinalResult(float accum) { if (applyFinalTransform()) { float f = accum / n(); this.finalResult = f; return f; } else { this.finalResult = accum; return accum; } }
@Override public float calculateFinalResult(float accum, long n) { if (applyFinalTransform()) return accum / n; return accum; }
@Override public String onnxName() { throw new NoOpNameFoundException("No onnx op opName found for " + opName()); }
@Override public INDArray amean(int... dimension) { return Nd4j.getExecutioner().exec(new AMean(this), dimension); }