Skip to content

Commit

Permalink
Merge pull request #8557 from husisy/patch-1
Browse files Browse the repository at this point in the history
Update functions.rst
  • Loading branch information
mergify[bot] committed Apr 16, 2020
2 parents 97e8327 + 0ae73d0 commit a0fc29c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/guides/functions.rst
Expand Up @@ -660,7 +660,7 @@ First, we have to define a function on variables:
return gx, gW, gb

def linear(x, W, b):
return LinearFunction()(x, W, b)
return LinearFunction().apply((x, W, b))

This function takes three arguments: input, weight, and bias.
It can be used as a part of model definition, though is inconvenient since the user have to manage the weight and bias parameters directly.
Expand Down

0 comments on commit a0fc29c

Please sign in to comment.