Compute the affine transform of a fully-connected layer: y = x @ W.T + b, where x is (batch, in_features), W is (out_features, in_features), and b is (out_features,).