Typical feed-forward layers map an input tensor x to Ax+b, for a matrix A and a vector b; both trainable. I\'m trying to create a feed-forward layer that maps an input tens