From e885264b5f2a29cb153b2d58b8acd301f69f2754 Mon Sep 17 00:00:00 2001
From: dupirefr <dupire.francois+pro@gmail.com>
Date: Sun, 16 Jun 2019 15:58:30 +0200
Subject: [PATCH] [BAEL-2996] Added ND4J dependency and matrices multiplication
 test

---
 libraries-2/pom.xml                           |  8 +++-
 .../com/baeldung/nd4j/INDArrayUnitTest.java   | 40 +++++++++++++++++++
 2 files changed, 47 insertions(+), 1 deletion(-)
 create mode 100644 libraries-2/src/test/java/com/baeldung/nd4j/INDArrayUnitTest.java

diff --git a/libraries-2/pom.xml b/libraries-2/pom.xml
index aca47f957d..bcb4f51bc2 100644
--- a/libraries-2/pom.xml
+++ b/libraries-2/pom.xml
@@ -55,6 +55,11 @@
             <artifactId>ejml-all</artifactId>
             <version>${ejml.version}</version>
         </dependency>
+		<dependency>
+			<groupId>org.nd4j</groupId>
+			<artifactId>nd4j-native</artifactId>
+			<version>${nd4j.version}</version>
+		</dependency>
 		<dependency>
 			<groupId>colt</groupId>
 			<artifactId>colt</artifactId>
@@ -124,7 +129,8 @@
 		<chronicle.map.version>3.17.2</chronicle.map.version>
         <crawler4j.version>4.4.0</crawler4j.version>
 		<spring-boot-starter.version>2.1.4.RELEASE</spring-boot-starter.version>
-		<ejml.version>0.37.1</ejml.version>
+		<ejml.version>0.38</ejml.version>
+		<nd4j.version>1.0.0-beta4</nd4j.version>
 		<colt.version>1.2.0</colt.version>
 		<la4j.version>0.6.0</la4j.version>
 	</properties>
diff --git a/libraries-2/src/test/java/com/baeldung/nd4j/INDArrayUnitTest.java b/libraries-2/src/test/java/com/baeldung/nd4j/INDArrayUnitTest.java
new file mode 100644
index 0000000000..a09eef9a57
--- /dev/null
+++ b/libraries-2/src/test/java/com/baeldung/nd4j/INDArrayUnitTest.java
@@ -0,0 +1,40 @@
+package com.baeldung.nd4j;
+
+import org.junit.jupiter.api.Test;
+import org.nd4j.linalg.api.ndarray.INDArray;
+import org.nd4j.linalg.factory.Nd4j;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class INDArrayUnitTest {
+
+    @Test
+    void givenTwoMatrices_whenMultiply_thenMultiplicatedMatrix() {
+        INDArray firstMatrix = Nd4j.create(
+          new double[][]{
+            new double[]{1d, 5d},
+            new double[]{2d, 3d},
+            new double[]{1d, 7d}
+          }
+        );
+
+        INDArray secondMatrix = Nd4j.create(
+          new double[][] {
+            new double[] {1d, 2d, 3d, 7d},
+            new double[] {5d, 2d, 8d, 1d}
+          }
+        );
+
+        INDArray expected = Nd4j.create(
+          new double[][] {
+            new double[] {26d, 12d, 43d, 12d},
+            new double[] {17d, 10d, 30d, 17d},
+            new double[] {36d, 16d, 59d, 14d}
+          }
+        );
+
+        INDArray actual = firstMatrix.mmul(secondMatrix);
+
+        assertThat(actual).isEqualTo(expected);
+    }
+}