fix v0.5 compatibility

rdeits · rdeits · commit c6499f601aae · 2017-05-04T11:54:45.000-04:00
diff --git a/src/layers/multinomial-logistic-loss.jl b/src/layers/multinomial-logistic-loss.jl
@@ -93,7 +93,7 @@ function forward(backend::CPUBackend, state::MultinomialLogisticLossLayerState,
 
   idx_all = map(1:length(dims)) do i
     if i == state.op_dim
-      round.(Int, label) + 1
+      map(x -> round(Int, x), label) .+ 1
     else
       dim = dims[i]
       reshape(1:dim, [j == i? dim : 1 for j = 1:length(dims)]...)
diff --git a/src/layers/softmax-loss.jl b/src/layers/softmax-loss.jl
@@ -56,7 +56,7 @@ function backward(backend::CPUBackend, state::SoftmaxLossLayerState, inputs::Vec
 
     idx_all = map(1:length(dims)) do i
       if i == state.logistic.op_dim
-        round.(Int, label) .+ 1
+        map(x -> round(Int, x), label) .+ 1
       else
         dim = dims[i]
         reshape(1:dim, [j == i? dim : 1 for j = 1:length(dims)]...)
