org.deeplearning4j.arbiter.layers.ActivationLayerSpace Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arbiter-deeplearning4j Show documentation

There is a newer version: 1.0.0-beta7

Show newest version

package org.deeplearning4j.arbiter.layers;

import org.deeplearning4j.arbiter.util.CollectionUtils;
import org.deeplearning4j.nn.conf.layers.ActivationLayer;

/**
 * Layer space for {@link ActivationLayer}
 *
 * @author Alex Black
 */
public class ActivationLayerSpace extends FeedForwardLayerSpace {
    protected ActivationLayerSpace(Builder builder) {
        super(builder);

        this.numParameters = CollectionUtils.countUnique(collectLeaves());
    }

    @Override
    public ActivationLayer getValue(double[] parameterValues) {
        ActivationLayer.Builder b = new ActivationLayer.Builder();
        super.setLayerOptionsBuilder(b, parameterValues);
        return b.build();
    }

    public static class Builder extends FeedForwardLayerSpace.Builder{

        @SuppressWarnings("unchecked")
        public ActivationLayerSpace build(){
            return new ActivationLayerSpace(this);
        }
    }

    @Override
    public String toString() {
        return toString(", ");
    }

    @Override
    public String toString(String delim) {
        return "ActivationLayerSpace(" + super.toString(delim) + ")";
    }
}