I am trying to build a neural network that takes in data in form of a matrix and outputs a vector but I don\'t know what layers to use to perform that. My current model is t