fix calculation bug

src/tf/tf.cpp

@@ -567,13 +567,13 @@ struct tf_parser
     instruction_ref
     parse_onehot(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
     {
-        auto indices       = args[0]->eval().get<int64_t>().to_vector();
+        auto indices       = args[0]->eval().get<int32_t>().to_vector();
         int depth          = args[1]->eval().at<int32_t>();
-        int axis           = -1;
+        int64_t axis           = -1;
         size_t num_indices = indices.size();
         float on_value     = args[2]->eval().at<float>();
         float off_value    = args[3]->eval().at<float>();
-        if(contains(attributes, "axis"))
+        if (contains(attributes, "axis"))
             axis = attributes.at("axis").i();
         if(axis == -1)
         {
@@ -582,8 +582,8 @@ struct tf_parser
             std::fill(output.begin(), output.end(), off_value);
             for(size_t i = 0; i < num_indices; i++)
             {
-                if(indices[i] >= 0 and indices[i] < num_indices)
-                    output[depth * i + indices[i]] = on_value;
+                if(indices[i] >= 0 and indices[i] < depth)
+                    output.at(depth * i + indices[i]) = on_value;
             }
             return prog.add_literal(s, output);
         }