コード例 #1
0
 protected void verifyElements(IComplexNDArray d, IComplexNDArray d2) {
   for (int i = 0; i < d.rows(); i++) {
     for (int j = 0; j < d.columns(); j++) {
       IComplexNumber test1 = d.getComplex(i, j);
       IComplexNumber test2 = d2.getComplex(i, j);
       assertEquals(
           test1.realComponent().doubleValue(), test2.realComponent().doubleValue(), 1e-6);
       assertEquals(
           test1.imaginaryComponent().doubleValue(),
           test2.imaginaryComponent().doubleValue(),
           1e-6);
     }
   }
 }
コード例 #2
0
 @Test
 public void testTranspose() {
   IComplexNDArray ndArray =
       Nd4j.createComplex(
           new double[] {
             1.0,
             0.0,
             2.0,
             0.0,
             3.0,
             0.0,
             4.0,
             0.0,
             5.0,
             0.0,
             6.0,
             0.0,
             6.999999999999999,
             0.0,
             8.0,
             0.0,
             9.0,
             0.0,
             10.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0,
             0.0
           },
           new int[] {16, 1});
   IComplexNDArray transposed2 = ndArray.transpose();
   assertEquals(getFailureMessage(), 16, transposed2.columns());
 }