diff --git a/brainpy/_src/dnn/linear.py b/brainpy/_src/dnn/linear.py index a1d31e08..a750142a 100644 --- a/brainpy/_src/dnn/linear.py +++ b/brainpy/_src/dnn/linear.py @@ -981,7 +981,7 @@ def __init__( class JitFPLinear(Layer): - def get_connect_matrix(self): + def get_conn_matrix(self): return bm.jitconn.get_conn_matrix(self.prob, self.seed, shape=(self.num_out, self.num_in), transpose=self.transpose,