Function TEST

src/opr/test/indexing.cpp:168–172

}  // anonymous namespace

TEST(TestOprDiag, Diag) {
    TensorShapeArray cases = {{7, 7}, {7, 9}, {9, 7}, {8}};
    for (int32_t k = -3; k < 3; ++k)
        test_diag(k, cases);
}

TEST(TestOprIndexing, OneHot2D) {
    TensorShapeArray cases = {{1, 1}, {2, 2}, {10, 8}, {8, 10}};