Use batch size consistently

#1
by dennis-rall - opened

I found it a bit confusing that the output is described in one place as (batch_size, num_features) and in another place as (1, num_features). I suggest sticking with one method and think that (batch_size,...) gives more information

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment