@Override public Op opForDimension(int index, int dimension) { INDArray xAlongDimension = x.vectorAlongDimension(index, dimension); Mean ret; if (y() != null) ret = new Mean(xAlongDimension, y.vectorAlongDimension(index, dimension), xAlongDimension.length()); else ret = new Mean(x.vectorAlongDimension(index, dimension)); ret.setApplyFinalTransform(applyFinalTransform()); return ret; }
@Override public Op opForDimension(int index, int... dimension) { INDArray xAlongDimension = x.tensorAlongDimension(index, dimension); Mean ret; if (y() != null) ret = new Mean(xAlongDimension, y.tensorAlongDimension(index, dimension), xAlongDimension.length()); else ret = new Mean(x.tensorAlongDimension(index, dimension)); ret.setApplyFinalTransform(applyFinalTransform()); return ret; }