Java tutorial
/* * * * Copyright 2015 Skymind,Inc. * * * * Licensed under the Apache License, Version 2.0 (the "License"); * * you may not use this file except in compliance with the License. * * You may obtain a copy of the License at * * * * http://www.apache.org/licenses/LICENSE-2.0 * * * * Unless required by applicable law or agreed to in writing, software * * distributed under the License is distributed on an "AS IS" BASIS, * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * * See the License for the specific language governing permissions and * * limitations under the License. * * */ package org.nd4j.linalg.api.ops.impl.transforms; import org.apache.commons.math3.util.FastMath; import org.nd4j.linalg.api.complex.IComplexNumber; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.api.ops.BaseTransformOp; import org.nd4j.linalg.api.ops.Op; import org.nd4j.linalg.util.ComplexUtil; /** * Log elementwise function * * @author Adam Gibson */ public class Log extends BaseTransformOp { public Log() { } public Log(INDArray x, INDArray z) { super(x, z); } public Log(INDArray x, INDArray z, long n) { super(x, z, n); } public Log(INDArray x, INDArray y, INDArray z, long n) { super(x, y, z, n); } public Log(INDArray x) { super(x); } @Override public int opNum() { return 5; } @Override public String name() { return "log"; } @Override public IComplexNumber op(IComplexNumber origin, double other) { return ComplexUtil.log(origin); } @Override public IComplexNumber op(IComplexNumber origin, float other) { return ComplexUtil.log(origin); } @Override public IComplexNumber op(IComplexNumber origin, IComplexNumber other) { return ComplexUtil.log(origin); } @Override public float op(float origin, float other) { return (float) FastMath.log(origin); } @Override public double op(double origin, double other) { return FastMath.log(origin); } @Override public double op(double origin) { return FastMath.log(origin); } @Override public float op(float origin) { return (float) FastMath.log(origin); } @Override public IComplexNumber op(IComplexNumber origin) { return ComplexUtil.log(origin); } @Override public Op opForDimension(int index, int dimension) { INDArray xAlongDimension = x.vectorAlongDimension(index, dimension); if (y() != null) return new Log(xAlongDimension, y.vectorAlongDimension(index, dimension), z.vectorAlongDimension(index, dimension), xAlongDimension.length()); else return new Log(xAlongDimension, z.vectorAlongDimension(index, dimension), x.lengthLong()); } @Override public Op opForDimension(int index, int... dimension) { INDArray xAlongDimension = x.tensorAlongDimension(index, dimension); if (y() != null) return new Log(xAlongDimension, y.tensorAlongDimension(index, dimension), z.tensorAlongDimension(index, dimension), xAlongDimension.length()); else return new Log(xAlongDimension, z.tensorAlongDimension(index, dimension), x.lengthLong()); } }