y = x @ W + b Is how you'd write it in NumPy and most (/all?) deep learning libraries, with x.shape=[batch, ndim] I'd personally prefer that to be the convent

y = x @ W + b Is how you'd write it in NumPy and most (/all?) deep learning libraries, with x.shape=[batch, ndim] I'd personally prefer that to be the convention, for consistency.