From e5837dc8b3ea47a9eda960bcf34f8823da804315 Mon Sep 17 00:00:00 2001 From: Yaohui Liu Date: Sun, 5 Mar 2023 02:05:19 +0800 Subject: [PATCH] Resolve the comment. --- .../Engine/DataAdapters/TensorLikeDataAdapter.cs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs b/src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs index f53c67c4..a7e1d7e3 100644 --- a/src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs +++ b/src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs @@ -21,7 +21,7 @@ namespace Tensorflow.Keras.Engine.DataAdapters { 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)));