org.nd4j.linalg.api.ops.impl.layers.recurrent.weights.RNNWeights Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation

There is a newer version: 1.0.0-M2.1

package org.nd4j.linalg.api.ops.impl.layers.recurrent.weights;

import java.util.Arrays;
import org.nd4j.autodiff.samediff.SDVariable;
import org.nd4j.linalg.util.ArrayUtil;

public abstract class RNNWeights {
    public abstract SDVariable[] args();

    protected static SDVariable[] filterNonNull(SDVariable... args){
        int count = 0;
        for(SDVariable v : args){
            if(v != null){
                count++;
            }
        }

        SDVariable[] res = new SDVariable[count];

        int i = 0;

        for(SDVariable v : args){
            if(v != null){
                res[i] = v;
                i++;
            }
        }

        return res;
    }

    public SDVariable[] argsWithInputs(SDVariable... inputs){
        return ArrayUtil.combine(inputs, args());
    }
}