output_concat = self.transpose_output(output)
return np.dot(output_concat, self.W_o) + self.b_o