TFTensor, add convenience boolean overloads

migueldeicaza · migueldeicaza · commit 6bb28e935968 · 2017-05-22T00:29:12.000-04:00
diff --git a/TensorFlowSharp/Tensor.cs b/TensorFlowSharp/Tensor.cs
@@ -233,6 +233,16 @@ public unsafe TFTensor (int value)
 			handle = TF_NewTensor (TFDataType.Int32, zeroDims: IntPtr.Zero, num_dims: 0, data: (IntPtr)v, len: (UIntPtr)sizeof (int), deallocator: FreeTensorData, deallocator_arg: IntPtr.Zero);
 		}
 
+		/// <summary>
+		/// Creates a constant tensor with a single dimension from a boolean value.
+		/// </summary>
+		public unsafe TFTensor (bool value)
+		{
+			var v = (bool*)Marshal.AllocHGlobal (sizeof (bool));
+			*v = value;
+			handle = TF_NewTensor (TFDataType.Bool, zeroDims: IntPtr.Zero, num_dims: 0, data: (IntPtr)v, len: (UIntPtr)sizeof (int), deallocator: FreeTensorData, deallocator_arg: IntPtr.Zero);
+		}
+
 		/// <summary>
 		/// Creates a constant tensor with a single dimension from an sbyte value.
 		/// </summary>
@@ -315,11 +325,16 @@ public unsafe TFTensor (long value)
 
 		// Convenience, should I add T[,] and T[,,] as more convenience ones?
 
+		/// <summary>
+		/// Creates a 1 dimensional tensor from an array of booleans.
+		/// </summary>
+		/// <param name="data">Data.</param>
+		public TFTensor (bool [] data) : base (SetupTensor (TFDataType.Bool, data, size: 1)) { }
 		/// <summary>
 		/// Creates a 1 dimensional tensor from an array of sbytes.
 		/// </summary>
 		/// <param name="data">Data.</param>
-		public TFTensor (sbyte [] data) : base (SetupTensor (TFDataType.Int8, data, size: 2)) { }
+		public TFTensor (sbyte [] data) : base (SetupTensor (TFDataType.Int8, data, size: 1)) { }
 		/// <summary>
 		/// Creates a 1 dimensional tensor from an array of bytes.
 		/// </summary>
@@ -453,6 +468,16 @@ public static implicit operator TFTensor (int value)
 			return new TFTensor (value);
 		}
 
+		/// <summary>
+		/// Converts a boolean into a 1-dimensional, 1-valued tensor.
+		/// </summary>
+		/// <returns>The tensor representing the integer value.</returns>
+		/// <param name="value">Value to initialize the tensor with.</param>
+		public static implicit operator TFTensor (bool value)
+		{
+			return new TFTensor (value);
+		}
+
 		/// <summary>
 		/// Converts a long into a 1-dimensional, 1-valued tensor.
 		/// </summary>
