Function predict

intermediate_source/jacobians_hessians.py:32–33

(weight, bias, x)

# This is a simple linear function with non-linear activation.

def predict(weight, bias, x):
    return F.linear(x, weight, bias).tanh()

######################################################################
# Let's add some dummy data: a weight, a bias, and a feature vector x.