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

Question

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

我的函数的伪代码是：

z1 = myGauss(arg, arg2)

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

Answer 1

如果你能够实现你对现有操作的组合所做的，那就太棒了！您不需要创建新操作。

在某些情况下，我们发现有必要创建新的操作系统，但是：

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

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