Skip to content

Commit e5837dc

Browse files
AsakusaRinneOceania2018
authored andcommitted
Resolve the comment.
1 parent 875063a commit e5837dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/TensorFlowNET.Keras/Engine/DataAdapters/TensorLikeDataAdapter.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ public TensorLikeDataAdapter(DataAdapterArgs args)
2121
{
2222
this.args = args;
2323
_process_tensorlike();
24-
num_samples = (int)args.X[0].shape[0];
24+
num_samples = (int)args.X.shape[0];
2525
var batch_size = args.BatchSize == -1 ? 32 : args.BatchSize;
2626
_batch_size = batch_size;
2727
_size = Convert.ToInt32(Math.Ceiling(num_samples / (batch_size + 0.0f)));

0 commit comments

Comments
 (0)