tf.rank SciSharp#396

Oceania2018 · Oceania2018 · commit 7f4ed1f0dbe1 · 2019-09-16T21:02:37.000-05:00
diff --git a/src/TensorFlowNET.Core/APIs/tf.array.cs b/src/TensorFlowNET.Core/APIs/tf.array.cs
@@ -84,6 +84,15 @@ public Tensor where<Tx, Ty>(Tensor condition, Tx x, Ty y, string name = null)
         public Tensor transpose<T1>(T1 a, int[] perm = null, string name = "transpose", bool conjugate = false)
             => array_ops.transpose(a, perm, name, conjugate);
 
+        /// <summary>
+        /// Returns the rank of a tensor.
+        /// </summary>
+        /// <param name="input"></param>
+        /// <param name="name"></param>
+        /// <returns>Returns a 0-D `int32` `Tensor` representing the rank of `input`.</returns>
+        public Tensor rank(Tensor input, string name = null)
+            => array_ops.rank(input, name: name);
+
         public Tensor squeeze(Tensor input, int[] axis = null, string name = null, int squeeze_dims = -1)
             => gen_array_ops.squeeze(input, axis, name);
 
