• Xavier Bouthillier's avatar
    Add handling of negative axis for Join and GpuJoin · 3b3a33f9
    Xavier Bouthillier 提交于
    Negative axis were automatically handled by numpy implementation inside
    Join.{perform,c_code}. However, both Join.grad and
    GpuJoin.{perform,c_code,grad} implementations were not thought to take
    care of negative axis.
    
    This commit adds checks for exceeding negative axis in Join.make_node for
    constants and in {Gpu}Join.{perform, c_code} for symbolic variables. Axis are
    converted to positive values inside GpuJoin.{perform, c_code} since current
    implementation does not support negative axis.
    3b3a33f9
basic.py 185.2 KB