data_flow.dynamic_partition

Author: Oceania2018

src/TensorFlowNET.Core/APIs/tf.data_flow.cs

@@ -27,7 +27,19 @@ public partial class tensorflow
         /// <param name="data"></param>
         /// <param name="name"></param>
         /// <returns></returns>
-        public static Tensor dynamic_stitch(Tensor[] indices, Tensor[] data, string name = null)
+        public Tensor dynamic_stitch(Tensor[] indices, Tensor[] data, string name = null)
             => gen_data_flow_ops.dynamic_stitch(indices, data, name: name);
+
+        /// <summary>
+        /// Partitions `data` into `num_partitions` tensors using indices from `partitions`.
+        /// </summary>
+        /// <param name="data"></param>
+        /// <param name="partitions"></param>
+        /// <param name="num_partitions">The number of partitions to output.</param>
+        /// <param name="name"></param>
+        /// <returns></returns>
+        public Tensor[] dynamic_partition(Tensor data, Tensor partitions, int num_partitions,
+            string name = null)
+            => gen_data_flow_ops.dynamic_partition(data, partitions, num_partitions, name: name);
     }
 }

src/TensorFlowNET.Core/Operations/gen_data_flow_ops.cs

@@ -27,6 +27,19 @@ public static Tensor dynamic_stitch(Tensor[] indices, Tensor[] data, string name
             return _op.output;
         }
 
+        public static Tensor[] dynamic_partition(Tensor data, Tensor partitions, int num_partitions, 
+            string name = null)
+        {
+            var _op = _op_def_lib._apply_op_helper("DynamicPartition", name, new 
+            { 
+                data,
+                partitions,
+                num_partitions
+            });
+
+            return _op.outputs;
+        }
+
         public static (Tensor, Tensor) tensor_array_v3<T>(T size, TF_DataType dtype = TF_DataType.DtInvalid, 
             TensorShape element_shape = null, bool dynamic_size = false, bool clear_after_read = true, 
             bool identical_element_shapes = false, string tensor_array_name = "", string name = null)