@Test
public void testPutComplex() {
INDArray fourTwoTwo = Nd4j.linspace(1, 16, 16).reshape(4, 2, 2);
IComplexNDArray test = Nd4j.createComplex(4, 2, 2);
for (int i = 0; i < test.vectorsAlongDimension(0); i++) {
INDArray vector = fourTwoTwo.vectorAlongDimension(i, 0);
IComplexNDArray complexVector = test.vectorAlongDimension(i, 0);
for (int j = 0; j < complexVector.length(); j++) {
complexVector.putReal(j, vector.getFloat(j));
}
}
for (int i = 0; i < test.vectorsAlongDimension(0); i++) {
INDArray vector = fourTwoTwo.vectorAlongDimension(i, 0);
IComplexNDArray complexVector = test.vectorAlongDimension(i, 0);
assertEquals(vector, complexVector.real());
}
}
