org.nd4j.linalg.transformation.Multiply Maven / Gradle / Ivy
package org.nd4j.linalg.transformation;
import org.nd4j.linalg.api.ndarray.INDArray;
public class Multiply implements MatrixTransform {
/**
*
*/
private static final long serialVersionUID = 6270130254778514061L;
@Override
public INDArray apply(INDArray input) {
return input.mul(input);
}
}