Apply Layer's logic to the inputs
@override Matrix act(dynamic inputs, {bool train = false}) { return Matrix.fromLists(inputs).T; }