tensor_stream 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 408dd69cf3211817d31734564f9a51c9ff9fdfca9ab7f207994f67b2b6d82303
-  data.tar.gz: f07f0b3ebc776dbd1e836968d5e8c97b9730519c6d57251f14cf7a08581b77a3
+  metadata.gz: a5d2223b3321554a5529fc7be45f6be485f66d0024d8b3ff4fadb8adfb20ba2a
+  data.tar.gz: 229532234896767058fc8d3dff0897650504345aaed1ce3d57b13d09f017f19e
 SHA512:
-  metadata.gz: '079a28b96e99180438059036db170dccbb8437188893031a607fcdebc8e66d6ea6678f1da83bd2bf2a8515434e061867e02f9342d00f5a8aadbceb141465ed09'
-  data.tar.gz: 1f8556098e4fa005a36e2884a079b9ea862aeef0c077623215430b69201a6e7e56b3fd4e33036f54a5c554cbf17fffe71a792f1984f4ac6fbcb116a14104453e
+  metadata.gz: '084f3d8fa7e74fccdbcbc42e004716520e9dc7a0cf7568ecbab8182e21ea0930ee58da9ca262b245e28e38df99e20f7dca029007c4e11030a3f6d62b4dd2e089'
+  data.tar.gz: a0c9e9987f557ed3cb301114c8f7bb6a91e1344dfeb96d72fbb4c61d850568e1d73ae8b060cabb655954a1d7b36379579eb221abfb955204454e28eae94cdd03

data/lib/tensor_stream/helpers/tensor_mixins.rb

@@ -1,8 +1,7 @@
 module TensorStream
   module TensorMixins
     def +(other)
-      TensorStream.check_data_types(self, other)
-      _op(:add, self, other)
+      _op(:add, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def [](index)
@@ -10,22 +9,18 @@ module TensorStream
     end
 
     def *(other)
-      TensorStream.check_data_types(self, other)
       _op(:mul, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def **(other)
-      TensorStream.check_data_types(self, other)
       _op(:pow, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def /(other)
-      TensorStream.check_data_types(self, other)
       _op(:div, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def -(other)
-      TensorStream.check_data_types(self, other)
       _op(:sub, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
@@ -67,43 +62,35 @@ module TensorStream
     end
 
     def <(other)
-      TensorStream.check_data_types(self, other)
-      _op(:less, self, other)
+      _op(:less, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def !=(other)
-      TensorStream.check_data_types(self, other)
-      _op(:not_equal, self, other)
+      _op(:not_equal, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def >(other)
-      TensorStream.check_data_types(self, other)
-      _op(:greater, self, other)
+      _op(:greater, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def >=(other)
-      TensorStream.check_data_types(self, other)
-      _op(:greater_equal, self, other)
+      _op(:greater_equal, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def <=(other)
-      TensorStream.check_data_types(self, other)
-      _op(:less_equal, self, other)
+      _op(:less_equal, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def and(other)
-      TensorStream.check_data_types(self, other)
-      _op(:logical_and, self, other)
+      _op(:logical_and, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def matmul(other)
-      TensorStream.check_data_types(self, other)
-      _op(:mat_mul, self, other)
+      _op(:mat_mul, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def dot(other)
-      TensorStream.check_data_types(self, other)
-      _op(:mat_mul, self, other)
+      _op(:mat_mul, self, TensorStream.convert_to_tensor(other, dtype: data_type))
     end
 
     def cast(data_type = :float32, name: nil)

data/lib/tensor_stream/operation.rb

@@ -100,6 +100,8 @@ module TensorStream
         :boolean
       when :shape, :rank, :shape_n
         options[:out_type] || :int32
+      when :zeros, :ones
+        options[:dtype] || :float32
       when :random_standard_normal, :random_uniform, :glorot_uniform, :truncated_normal
         passed_data_type || :float32
       when :concat

data/lib/tensor_stream/version.rb

@@ -1,5 +1,5 @@
 module TensorStream
-  VERSION = "1.0.3".freeze
+  VERSION = "1.0.4".freeze
 
   def self.version
     VERSION

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tensor_stream
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Joseph Emmanuel Dayo
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-03-03 00:00:00.000000000 Z
+date: 2019-03-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler