diff options
| author | txh18 <[email protected]> | 2015-05-26 15:14:37 +0800 |
|---|---|---|
| committer | txh18 <[email protected]> | 2015-05-26 15:14:37 +0800 |
| commit | ad6fa788f1b2e88167407922f18118cf905eecaf (patch) | |
| tree | ea8a8e0753ed8f49b37c475c4526a313e371568e | |
| parent | 676eb07afc1307c38c3b0089476f7cabe17cc72e (diff) | |
added nerv_matrix_(get_dataref_value) and tested in examples/mmatrix_example, however it is a little strange to return 101
| -rw-r--r-- | examples/mmatrix_example.lua | 1 | ||||
| -rw-r--r-- | matrix/generic/matrix.c | 6 |
2 files changed, 7 insertions, 0 deletions
diff --git a/examples/mmatrix_example.lua b/examples/mmatrix_example.lua index 679fde3..53b7fed 100644 --- a/examples/mmatrix_example.lua +++ b/examples/mmatrix_example.lua @@ -10,6 +10,7 @@ for i = 0, m - 1 do dm[i][j] = t end end +print("test fm:get_dataref_value:", fm:get_dataref_value()) print(fm) -- print(fm:softmax()) print(dm) diff --git a/matrix/generic/matrix.c b/matrix/generic/matrix.c index 417c534..c3838d2 100644 --- a/matrix/generic/matrix.c +++ b/matrix/generic/matrix.c @@ -122,12 +122,18 @@ static int nerv_matrix_(nrow)(lua_State *L) { return 1; } +static int nerv_matrix_(get_dataref_value)(lua_State *L) { + Matrix *self = luaT_checkudata(L, 1, nerv_matrix_(tname)); + lua_pushinteger(L, *(self->data_ref)); + return 1; +} static const luaL_Reg nerv_matrix_(methods)[] = { {"get_elem", nerv_matrix_(get_elem)}, {"set_elem", nerv_matrix_(set_elem)}, {"ncol", nerv_matrix_(ncol)}, {"nrow", nerv_matrix_(nrow)}, + {"get_dataref_value", nerv_matrix_(get_dataref_value)}, {"__index__", nerv_matrix_(index)}, {"__newindex__", nerv_matrix_(newindex)}, {NULL, NULL} |