tensor_stream 0.9.6 → 0.9.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4059c6bb82fc83c5aee32090e9c58231f9fd421afb3aa8bea1cbff4857edeb18
-  data.tar.gz: b7ceee8509455146402c3b3cdf9fa0a63eeec3dfee5512dd6e097221f1247481
+  metadata.gz: 1f312e783bb684267b34aa4a12cfcea525095f0fa7237c4666c19e04e21b669c
+  data.tar.gz: 8910619b66cff18f07dd6c4c0ff0a68273f4534e702ef2ace26dfb99de162850
 SHA512:
-  metadata.gz: 329eb6875b1add417656537beac95d67c1c061fa3eba3b04031d3ac6f1a20f7e26bf12e66873b70ad575fe7d9f0482c02888f0b544405ba9d73d5b325070966f
-  data.tar.gz: 639a4d90dc1525e063bcdac3b30805a38697b009634d17cb745c99189b88b10b95ce49a9136d5c9c0617fbf3ca3ee1a74429a8579437b7896954ff6529952e79
+  metadata.gz: 6586f8f501505322fe94da573d38923fac7546c72b9a96df4bac22e9c2ec3bb2b60fa1682fb8ed330a58d068a6917e829a9f69ab6a8be4f8a2b6d3e46d7891f6
+  data.tar.gz: 8fbf3fb157c6c30240a5bc9d578341c38d562f7e2b6510640e4d1e568770954275a6f21c0c53e310cd665c8773ed7432c0a16338bf806cd4e6cdcbdea48a61a1

data/CHANGELOG.md

@@ -4,6 +4,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.9.6] - 2018-11-18
+- [NEW OP] Convolutional networks - conv2d, conv2d_backprop_filter, conv2d_backprop_input
+- [IMAGE] Exposed image resampling options
+- [BUG FIX] fix argmin, argmax handling of NaN values
+
 ## [0.9.5] - 2018-11-05
 - [NEW OP] assert_equal, relu6
 - [TRAINING] learning_rate_decay, dropout

data/lib/tensor_stream/evaluator/ruby/nn_ops.rb

@@ -264,6 +264,8 @@ module TensorStream
 
             (0...height).step(height_stride).each do |y|
               (0...width).step(width_stride).each do |x|
+                img_grad = grad[b][y/height_stride][x/width_stride]
+
                 (0...f_height).each do |f_y|
                   (0...f_width).each do |f_x|
                     next if x + f_x >= width
@@ -271,7 +273,7 @@ module TensorStream
 
                     channels.times.each do |c|
                       image_gradient[y + f_y][x + f_x][c] += Array.new(output_channels) do |o_c|
-                        filter[f_y][f_x][c][o_c] * grad[b][(y/height_stride) + f_y][(x/width_stride) + f_x][o_c]
+                        filter[f_y][f_x][c][o_c] * img_grad[o_c]
                       end.reduce(:+)
                     end
                   end
@@ -298,15 +300,15 @@ module TensorStream
 
             (0...height).step(height_stride).each do |y|
               (0...width).step(width_stride).each do |x|
+                image_grad = grad[index][y/height_stride][x/width_stride]
                 filter_result = (0...f_height).map do |f_y|
                   (0...f_width).map do |f_x|
-
                     next Array.new(input_channels * output_channels) { 0.0 } if x + f_x >= width
                     next Array.new(input_channels * output_channels) { 0.0 } if y + f_y >= height
 
                     image[y + f_y][x + f_x].each_with_index.map do |image_channel, c_channel|
                       output_channels.times.map do |o_c|
-                        image_channel * grad[index][(y/height_stride) + f_y][(x/width_stride) + f_x][o_c]
+                        image_channel * image_grad[o_c]
                       end
                     end
                   end

data/lib/tensor_stream/helpers/infer_shape.rb

@@ -178,6 +178,10 @@ module TensorStream
         new_shape = tensor.inputs[0].shape.shape.dup
         new_shape[3] = tensor.inputs[1].shape.shape[3]
         new_shape
+      when :conv2d_backprop_input
+        return nil unless tensor.inputs[0].value
+
+        tensor.inputs[0].value
       else
         return nil if tensor.inputs[0].nil?
         return tensor.inputs[0].shape.shape if tensor.inputs.size == 1

data/lib/tensor_stream/operation.rb

@@ -66,6 +66,8 @@ module TensorStream
         passed_data_type || :float32
       when :concat
         @inputs[1].data_type
+      when :conv2d_backprop_input
+        @inputs[1].data_type
       when :index
         if @inputs[0].is_a?(ControlFlow)
 

data/lib/tensor_stream/version.rb

@@ -1,5 +1,5 @@
 module TensorStream
-  VERSION = '0.9.6'.freeze
+  VERSION = '0.9.7'.freeze
 
   def self.version
     VERSION

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tensor_stream
 version: !ruby/object:Gem::Version
-  version: 0.9.6
+  version: 0.9.7
 platform: ruby
 authors:
 - Joseph Emmanuel Dayo
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-11-18 00:00:00.000000000 Z
+date: 2018-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler