Resolve the comment.

SciSharp · Mar 4, 2023 · e5837dc · e5837dc
1 parent 875063a
commit e5837dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs b/src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs
@@ -21,7 +21,7 @@ public TensorLikeDataAdapter(DataAdapterArgs args)
         {
             this.args = args;
             _process_tensorlike();
-            num_samples = (int)args.X[0].shape[0];
+            num_samples = (int)args.X.shape[0];
             var batch_size = args.BatchSize == -1 ? 32 : args.BatchSize;
             _batch_size = batch_size;
             _size = Convert.ToInt32(Math.Ceiling(num_samples / (batch_size + 0.0f)));