Ejemplos de IComplexNDArray.assign en Java

Ejemplo n.º 1

Archivo: ComplexNDArrayTestsFortran.java Proyecto: ForeverAct/nd4j

  @Test
  public void testAssignOffset() {
    IComplexNDArray arr = Nd4j.complexOnes(5, 5);
    IComplexNDArray row = arr.slice(1);
    row.assign(1);
    assertEquals(Nd4j.complexOnes(5), row);

    IComplexNDArray row2 = arr.slice(2);
    row2.assign(Nd4j.complexValueOf(5, 2));
    assertEquals(Nd4j.complexValueOf(5, 2), row2);
  }