Function _make_conn

test/network/test_mstdp_florian.py:91–102

(npre, npost, rule, batch=1, nu=1e-2, dt=1.0, **kw)

# Helpers to drive a Connection with explicit, controlled spike trains.       #
# --------------------------------------------------------------------------- #
def _make_conn(npre, npost, rule, batch=1, nu=1e-2, dt=1.0, **kw):
    src = Input(n=npre)
    tgt = LIFNodes(n=npost)
    src.batch_size = batch
    tgt.batch_size = batch
    src.s = torch.zeros(batch, npre, dtype=torch.bool)
    tgt.s = torch.zeros(batch, npost, dtype=torch.bool)
    conn = Connection(src, tgt, nu=nu, update_rule=rule, reduction=torch.sum, **kw)
    conn.dt = dt
    with torch.no_grad():
        conn.w.zero_()
    return conn


def _drive(conn, pre, post, rewards):