ruby-cntk 0.1.0.pre1

data/ext/cntk/extconf.rb

@@ -0,0 +1,21 @@
+require "rbconfig"
+
+#RbConfig::MAKEFILE_CONFIG["CXX"] = "clang++-3.8"
+
+require 'mkmf'
+
+
+
+
+have_library("c++") or have_library("stdc++")
+
+# have_library("cntklibrary-2.0", nil, nil, " -L /cntk/build/cpu/release/lib/ ")
+dir_config("cntklibrary-2.0")
+have_library("cntklibrary-2.0")
+
+# rake2.3 compile -- --with-cntklibrary-2.0-lib=/cntk/build/cpu/release/lib/ --with-cntklibrary-2.0-include=/cntk/Source/CNTKv2LibraryDll/API/
+# rake2.3 compile -- --with-cntklibrary-2.0-lib=/cntk/cntk/lib/ --with-cntklibrary-2.0-include=/cntk/Include/
+$CXXFLAGS = ($CXXFLAGS || "") + " -std=c++11 -O2 -DSWIG " 
+# $LDFLAGS = ($LDFLAGS || "") + ""
+
+create_makefile('cntk/CNTK')

data/lib/cntk/function.rb

@@ -0,0 +1,68 @@
+module CNTK
+  class Function
+
+    def call(args)
+      if args.outputs.length == 1
+        return replace_placeholders({placeholders[0] => args.output})
+      else
+        raise "not implemented"
+      end
+    end
+
+    def forward(*args)
+      if args.length > 1
+        return __forward__(*args)
+      elsif args.length == 1
+        input = convert_to_value(args[0])
+        out = StdUMapVariableValue.new()
+        outputs().each{|o|
+          v = NDArrayView.new(CNTK::DataType_Double, 
+                              required_output_shape(o),
+                              required_output_buf(o),
+                              CNTK::DeviceDescriptor.default_device(),
+                              true)
+          out[o] = Value.new(v)
+        }
+        b = __forward__(input, out)
+        out = remove_dynamic_axes(out)
+        return [out, b]
+      end
+    end
+
+    def convert_to_value(h)
+      input = {}
+      h.each_pair{|k,val|
+        if val.respond_to?(:row_major?)
+          input[k] = Value.create(val)
+        else
+          input[k] = val
+        end
+      }
+      return input
+    end
+
+    #FIXME
+    # we must add dynamic axes?
+    def required_output_shape(ov)
+      ov.shape().to_a + [1,1]
+    end
+
+    def required_output_buf(ov)
+      [1.0] * ov.shape.total_size
+    end
+
+    def remove_dynamic_axes(out)
+      out1 = {}
+      out.each{|o,ov|
+        if ov.shape.rank == o.shape.rank + 2 and ov.shape.to_a[-2..-1] == [1,1]
+          out1[o] = ov.reshape( ov.shape.to_a[0..-3] )
+        else
+          out1[o] = ov
+        end
+      }
+      return out1
+    end
+
+    private :__forward__
+  end
+end

data/lib/cntk/ndarrayview.rb

@@ -0,0 +1,27 @@
+module CNTK
+  class NDArrayView
+
+    def self.create(a)
+      if a.respond_to?(:shape) and a.respond_to?(:row_major?)
+        if a.row_major?
+          # NDArrayView is column-major.
+          # So we must transpose a.
+          ta = a.transpose
+        end
+        return self.new(DataType_Double, a.shape, ta.flatten.to_a, 
+                        CNTK::DeviceDescriptor.default_device(), false)
+      else
+        raise "not implemented"
+      end
+    end
+
+    def to_narray
+      ret = Numo::DFloat[*to_vec()]
+      # NDArrayView is column-major and NArray is row-major.
+      # So we must reverse shape and transpose it.
+      ret = ret.reshape(*shape().reverse)
+      return ret.transpose
+    end
+
+  end
+end

data/lib/cntk/ndshape.rb

@@ -0,0 +1,10 @@
+module CNTK
+  class NDShape
+    alias :to_ary :dimensions
+    alias :to_a   :dimensions
+    
+    def reverse
+      to_ary.reverse
+    end
+  end
+end

data/lib/cntk/value.rb

@@ -0,0 +1,18 @@
+module CNTK
+  class Value
+
+    def self.create(a)
+      new(NDArrayView.create(a))
+    end
+
+    def to_narray
+      data().to_narray
+    end
+
+    def reshape(a)
+      na = to_narray().reshape(*a)
+      self.class.create(na)
+    end
+
+  end
+end

data/lib/cntk.rb

@@ -0,0 +1,5 @@
+require "cntk/CNTK"
+require "cntk/function"
+require "cntk/ndshape"
+require "cntk/ndarrayview"
+require "cntk/value"

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification
+name: ruby-cntk
+version: !ruby/object:Gem::Version
+  version: 0.1.0.pre1
+platform: ruby
+authors:
+- Takashi Tamura
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-02-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.5
+description: Ruby bindings for Microsoft  Cognitive Toolkit (CNTK), an open source
+  deep-learning toolkit
+email: ''
+executables: []
+extensions:
+- ext/cntk/extconf.rb
+extra_rdoc_files: []
+files:
+- LICENSE
+- ext/cntk/cntk_wrap.cxx
+- ext/cntk/extconf.rb
+- lib/cntk.rb
+- lib/cntk/function.rb
+- lib/cntk/ndarrayview.rb
+- lib/cntk/ndshape.rb
+- lib/cntk/value.rb
+homepage: https://github.com/tamuratak/ruby-cntk
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options:
+- "--exclude=."
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">"
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements:
+- CNTK >= 2.0b11
+rubyforge_project: 
+rubygems_version: 2.5.2
+signing_key: 
+specification_version: 4
+summary: Ruby bindings for Microsoft CNTK, an open source deep-learning toolkit
+test_files: []