tf.random_shuffle SciSharp#396

Oceania2018 · Oceania2018 · commit f3786406181a · 2019-09-17T23:29:43.000-05:00
diff --git a/src/TensorFlowNET.Core/APIs/tf.random.cs b/src/TensorFlowNET.Core/APIs/tf.random.cs
@@ -49,5 +49,18 @@ public Tensor truncated_normal(int[] shape,
             int? seed = null,
             string name = null)
             => random_ops.truncated_normal(shape, mean, stddev, dtype, seed, name);
+
+        /// <summary>
+        /// Randomly shuffles a tensor along its first dimension.
+        /// </summary>
+        /// <param name="value"></param>
+        /// <param name="seed"></param>
+        /// <param name="name"></param>
+        /// <returns>
+        /// A tensor of same shape and type as value, shuffled along its 
+        /// first dimension.
+        /// </returns>
+        public Tensor random_shuffle(Tensor value, int? seed = null, string name = null)
+            => random_ops.random_shuffle(value, seed: seed, name: name);
     }
 }
diff --git a/src/TensorFlowNET.Core/Operations/gen_random_ops.py.cs b/src/TensorFlowNET.Core/Operations/gen_random_ops.py.cs
@@ -90,6 +90,23 @@ public static Tensor random_uniform(Tensor shape, TF_DataType dtype, int? seed =
             return _op.outputs[0];
         }
 
+        /// <summary>
+        /// 
+        /// </summary>
+        /// <param name="value"></param>
+        /// <param name="seed"></param>
+        /// <param name="seed2"></param>
+        /// <param name="name"></param>
+        /// <returns></returns>
+        public static Tensor random_shuffle(Tensor value, int seed = 0, int seed2 = 0, string name = null)
+        {
+            var _op = _op_def_lib._apply_op_helper("RandomShuffle",
+                name: name,
+                args: new { value, seed, seed2 });
+
+            return _op.output;
+        }
+
         /// <summary>
         /// Outputs random values from a truncated normal distribution.
         /// </summary>
diff --git a/src/TensorFlowNET.Core/Operations/random_ops.py.cs b/src/TensorFlowNET.Core/Operations/random_ops.py.cs
@@ -104,6 +104,19 @@ public static Tensor random_uniform(Tensor shape,
             });
         }
 
+        /// <summary>
+        /// Randomly shuffles a tensor along its first dimension.
+        /// </summary>
+        /// <param name="value"></param>
+        /// <param name="seed"></param>
+        /// <param name="name"></param>
+        /// <returns></returns>
+        public static Tensor random_shuffle(Tensor value, int? seed = null, string name = null)
+        {
+            var (seed1, seed2) = random_seed.get_seed(seed);
+            return gen_random_ops.random_shuffle(value, seed: seed1.Value, seed2: seed2.Value, name: name);
+        }
+
         public static Tensor truncated_normal(int[] shape,
             float mean = 0.0f,
             float stddev = 1.0f,
