org.nd4j.examples.ElementWiseOperationExample Maven / Gradle / Ivy

Go to download
package org.nd4j.examples;

import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;

/**
 * Created by cvn on 9/6/14.
 */
 
public class ElementWiseOperationExample {

    public static void main(String[] args) {

        INDArray nd = Nd4j.create(new float[]{1,2,3,4},new int[]{2,2});
        INDArray nd2 = Nd4j.create(new float[]{5,6},new int[]{2,1}); //vector as column
        INDArray nd3 = Nd4j.create(new float[]{5,6},new int[]{2}); //vector as row
        INDArray nd4 = Nd4j.create(new float[]{5,6,7,8},new int[]{2,2});

        // Show initial matrices

        System.out.println(nd);
        System.out.println(nd2);
        System.out.println(nd3);
        System.out.println(nd4);

        //create nd-array variable ndv to be able to print result of nondestructive operations. add scalar to matrix and assign ndv the sum.

        INDArray ndv = nd.add(1);

        System.out.println(ndv);

        ndv = nd.mul(5);
        System.out.println(ndv);

        ndv = nd.sub(3);
        System.out.println(ndv);

        ndv = nd.div(2);
        System.out.println(ndv);

        //add column vector to matrix

        ndv = nd.addColumnVector(nd2);

        System.out.println(ndv);

        // add row vector to matrix

        ndv = nd.addRowVector(nd3);

        System.out.println(ndv);

        //multiply two matrices of equal dimensions elementwise.

        ndv = nd.mul(nd4);

        System.out.println(ndv);

    }

}