我什么时候应该定义一个新的TensorFlow操作？

Question

对于我的应用程序,我只能使用预定义的操作创建一个新功能.在这种情况下是否需要定义新的操作？

我的函数的伪代码是:

z1 = myGauss(arg, arg2)

def myGauss(arg, arg2):
  # Here I only used defined tensorflow operations

Answer 1

如果你能够实现你对现有操作组合的要求,那就太棒了!你并不需要创建一个新的运算.

在某些情况下,我们发现有必要创建一个新操作,但是:

• 有时,您可以通过将 ops融合到一个操作中来获得性能.例如,许多"训练"操作都有融合实现,即使它们最初是使用简单的操作实现的.

• 另一个例子是当你想为ops的组合定义一个渐变时(因为将表达式作为一个整体来考虑它更有效或更稳定).这是操作的基本原理tf.nn.softmax_cross_entropy_with_logits().