实现torch的__len__元函数

在我们的torch-dataframe项目中,我们尝试以如下方式实现__len__元函数:

MyClass.__len__ = argcheck{
    {name="self", type="MyClass"},
    {name="other", type="MyClass"},
    call=function(self, other)
    return self.n_rows
end}

这在Lua 5.2和5.3中可以工作,但是对于Lua 5.1、luajit 2.0和2.1,返回的变量不是实际行数,而是0。感觉它返回了一个新的MyClass实例,但是很难理解为什么。这里有一个关于__len更改的注释here,但那是我们目前找到的最好的文档提示。

稍有惊讶的是需要两个参数。当argcheck提供单个参数版本时:

MyClass.__len__ = argcheck{
   {name = "self", type = "MyClass"},
   call=function(self)
   return self.n_rows
end}

它会抛出:

[string "argcheck"]:28:
Arguments:

({
   self = MyClass  --
})

Got: MyClass, MyClass

我们目前依赖于argcheck重载函数的运算符来处理这个问题:

MyClass.__len__ = argcheck{
    {name="self", type="MyClass"},
    {name="other", type="MyClass"},
    call=function(self, other)
    return self.n_rows
end}

MyClass.__len__ = argcheck{
    overload=MyClass.__len__,
    {name="self", type="MyClass"},
    call=function(self)
    return self.n_rows
end}

更多细节,请看完整的类和travis报告:

测试案例

这是一个完整的测试案例,在5.2和5.3中按预期工作,或许更简洁地说明了问题而非完整的包:

require 'torch'
local argcheck = require "argcheck"

local MyClass = torch.class("MyClass")

function MyClass:init()
    self.n_rows = 0
end

MyClass.__len__ = argcheck{
    {name = "self", type = "MyClass"},
    {name = "other", type = "MyClass"},
    call=function(self, other)
    print(self.n_rows)
    print(other.n_rows)
    return(self.n_rows)
end}

local obj = MyClass.new()
obj.n_rows = 1
local n = #obj
print(n)

这将按预期打印:

1
1
1
点赞
用户409508
用户409508

这个问题与这个 SO 问题有关。在5.1中,对于它没有任何支持

表中的 __len 排定在 5.2 中得到支持。请参见 LuaFiveTwo。

2016-07-20 20:11:41