TensorShape of variable. #138

Author: Oceania2018

src/TensorFlowNET.Core/Tensors/tensor_util.cs

@@ -89,6 +89,11 @@ public static TensorShapeProto as_shape(long[] dims)
             return shape;
         }
 
+        public static TensorShape to_shape(long[] dims)
+        {
+            return new TensorShape(dims.Select(x => (int)x).ToArray());
+        }
+
         public static TensorShape as_shape(this IShape shape, int[] dims)
         {
             return new TensorShape(dims);

src/TensorFlowNET.Core/Variables/RefVariable.cs

@@ -18,6 +18,7 @@ public partial class RefVariable : VariableV1
         public Operation initializer => _initializer_op;
         public Operation op => _variable.op;
         public TF_DataType dtype => _variable.dtype;
+        public TensorShape shape => tensor_util.to_shape(_variable.shape);
 
         public string name => _variable.name;