org.nd4j.autodiff.samediff.api.OutAndGrad 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.autodiff.samediff.api;

import lombok.AllArgsConstructor;
import lombok.Data;
import org.nd4j.linalg.api.ndarray.INDArray;

import java.util.Map;

/**
 * A simple object holding two maps - one of output arrays, another of gradient arrays
 */
@AllArgsConstructor
@Data
public class OutAndGrad {

    private final Map outputs;
    private final Map gradients;

}