tfp.experimental.math.patch_manual_special_functions
Stay organized with collections
Save and categorize content based on your preferences.
Patches in the manually implemented special functions.
@contextlib.contextmanager
tfp.experimental.math.patch_manual_special_functions()
Normally you'd just use functions coming from the array library you're using,
but on some platforms (like TPU) the default implementations are
insufficiently precise for certain tasks when running under 32 bits (64 bit
implementations are typically okay).
This patches in manual implementations of those functions which are provide
higher precision at the cost of speed. The list of affected functions is:
explogexpm1log1plogsumexp (aka reduce_logsumexp)softplus
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2023-11-21 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2023-11-21 UTC."],[],[]]
View source on GitHub