tf.reduce_prod, tf.add_to_collections, tf.identity. SciSharp#396

Oceania2018 · Oceania2018 · commit c2138b20abc4 · 2019-09-18T13:35:31.000-05:00
diff --git a/src/TensorFlowNET.Core/APIs/tf.array.cs b/src/TensorFlowNET.Core/APIs/tf.array.cs
@@ -94,6 +94,15 @@ public Tensor expand_dims(Tensor input, int axis = -1, string name = null, int d
         public Tensor fill<T>(Tensor dims, T value, string name = null)
             => gen_array_ops.fill(dims, value, name: name);
 
+        /// <summary>
+        /// Return a tensor with the same shape and contents as input.
+        /// </summary>
+        /// <param name="input"></param>
+        /// <param name="name"></param>
+        /// <returns></returns>
+        public static Tensor identity(Tensor input, string name = null)
+            => array_ops.identity(input, name: name);
+
         /// <summary>
         /// Gather slices from params axis axis according to indices.
         /// </summary>
diff --git a/src/TensorFlowNET.Core/APIs/tf.math.cs b/src/TensorFlowNET.Core/APIs/tf.math.cs
@@ -414,6 +414,17 @@ public Tensor reduce_any(Tensor input_tensor, int axis = 0, bool keepdims = fals
         public Tensor reduce_all(Tensor input_tensor, int[] axis = null, bool keepdims = false, string name = null)
             => math_ops.reduce_all(input_tensor, axis: axis, keepdims: keepdims, name: name);
 
+        /// <summary>
+        /// Computes the product of elements across dimensions of a tensor.
+        /// </summary>
+        /// <param name="input_tensor"></param>
+        /// <param name="axis"></param>
+        /// <param name="keepdims"></param>
+        /// <param name="name"></param>
+        /// <returns></returns>
+        public Tensor reduce_prod(Tensor input_tensor, int[] axis = null, bool keepdims = false, string name = null)
+            => math_ops.reduce_prod(input_tensor, axis: axis, keepdims: keepdims, name: name);
+
         /// <summary>
         /// Computes the sum of elements across dimensions of a tensor.
         /// </summary>
diff --git a/src/TensorFlowNET.Core/APIs/tf.ops.cs b/src/TensorFlowNET.Core/APIs/tf.ops.cs
@@ -14,10 +14,18 @@ You may obtain a copy of the License at
    limitations under the License.
 ******************************************************************************/
 
+using System.Collections.Generic;
+
 namespace Tensorflow
 {
     public partial class tensorflow
     {
+        public void add_to_collection<T>(string name, T value)
+            => get_default_graph().add_to_collection(name, value);
+
+        public void add_to_collections<T>(List<string> names, T value)
+            => get_default_graph().add_to_collections(names, value);
+
         public Tensor assign(Tensor @ref, object value, bool validate_shape = true, bool use_locking = true, string name = null) 
             => state_ops.assign(@ref, value, validate_shape, use_locking, name);
 
