Function _drive_shaped

test/network/test_mstdp_florian.py:247–260

Drive a conv/local connection with explicitly-shaped spike trains. pre/post: [T, batch, *node_shape]. Returns (weight_change, [eligibility...]).

(conn, pre, post)

def _drive_shaped(conn, pre, post):
    """Drive a conv/local connection with explicitly-shaped spike trains.

    pre/post: [T, batch, *node_shape]. Returns (weight_change, [eligibility...]).
    """
    T = pre.shape[0]
    w0 = conn.w.detach().clone()
    eligs = []
    for t in range(T):
        conn.source.s = pre[t].bool()
        conn.target.s = post[t].bool()
        conn.update(reward=1.0)
        eligs.append(conn.update_rule.eligibility.detach().clone())
    return conn.w.detach() - w0, eligs


_LOCAL_CASES = [