Function TEST

test/iota.cpp:105–116

using af::iota;

TEST(Iota, CPP) {
    dim4 idims(23, 15, 1, 1);
    dim4 tdims(2, 2, 1, 1);
    dim4 fulldims;
    for (unsigned i = 0; i < 4; i++) { fulldims[i] = idims[i] * tdims[i]; }

    array output = iota(idims, tdims);
    array tileArray =
        tile(moddims(range(dim4(idims.elements()), 0), idims), tdims);

    ASSERT_ARRAYS_EQ(tileArray, output);
}