lib/dnn/core/embedding.rb in ruby-dnn-0.13.2 vs lib/dnn/core/embedding.rb in ruby-dnn-0.13.3
- old
+ new
@@ -26,10 +26,10 @@
[forward(input), Link.new(nil, self)]
end
def build
@built = true
- @weight = Param.new(Xumo::SFloat.new(@input_length), 0)
+ @weight = Param.new(Xumo::SFloat.new(@input_length), Xumo::SFloat[0])
@weight_initializer.init_param(self, @weight)
@weight_regularizer.param = @weight if @weight_regularizer
end
def forward(x)