array join(const int dim, const array& first, const array& second) { af_array out = 0; AF_THROW(af_join(&out, dim, first.get(), second.get())); return array(out); }
#include "../utils.h" #include "../template/out_in.h" #include "../template/out_in2.h" #include "../template/out_in3.h" static const struct luaL_Reg move_reorder_methods[] = { OUTIN(flat), OUTIN_ARG(flip, unsigned), { "af_join", [](lua_State * L) { lua_settop(L, 3); // dim, first, second af_array * arr_ud = NewArray(L);// dim, first, second, arr_ud af_err err = af_join(arr_ud, I(L, 1), GetArray(L, 2), GetArray(L, 3)); return PushErr(L, err); // dim, first, second, err, arr_ud } }, { "af_join_many", [](lua_State * L) { lua_settop(L, 3); // num, narrays, inputs af_array * arr_ud = NewArray(L);// num, narrays, inputs, arr_ud unsigned count = U(L, 2); std::vector<af_array> arrays; for (unsigned i = 1; i <= count; ++i)