• Michael Yang's avatar
    ml: update Context.Forward interface · 3e8b8a19
    Michael Yang authored
    update Context.Forward to accept multiple tensors to match
    Context.Compute signature
    
    update Context.Forward to return Context such that it can be chained
    with Context.Compute
    3e8b8a19
causal_test.go 16.7 KB