时间戳

首页 > 解决方案 > 我可以使用 Pytorch 或 Tensorflow 获得推导的表达式吗？

tensorflow - 我可以使用 Pytorch 或 Tensorflow 获得推导的表达式吗？

问题描述

我的问题是我是否可以在使用 PyTorch 或 TensorFlow 计算推导后获得表达式。为了更好地理解我的问题，我在下面提供了一个示例：

我有一个函数（x1、x2 和 x3 都是决策变量）

y = x1 + x1*x2 + x1*x3

y 根据 x1 的推导应该是：

1 + x2 + x3

我的问题是，如果我使用 PyTorch 或 TensorFlow 进行计算，我可以打印出“1 + x2 + x3”的表达式吗？非常感谢无论使用 PyTorch 还是 TensorFlow 都提供代码。

标签： tensorflowpytorchsymbolic-math

解决方案

我不知道在Pytorchor中是否可行Tensorflow，但你绝对可以使用SymPypython 库来做到这一点：

import sympy
x1 = sympy.symbols('x1')
x2 = sympy.symbols('x2')
x3 = sympy.symbols('x3')

y = x1 + x1*x2 + x1*x3
res = sympy.diff(y, x1)

print(res)

输出：

x2 + x3 + 1

推荐阅读