什么是 TensorSpec?(张量流 2.0)

数据挖掘 神经网络 深度学习 张量流 python-3.x
2022-03-10 00:59:06

什么是tf.TensorSpec()以及它的用途是什么?

1个回答

tf.function 主要使用 TensorSpec 来指定输入签名。tf.function 将为不同的输入形状和数据类型创建一个图,但您的函数图可能与不同的形状兼容。作为性能优化,您可以选择提供一个签名,这样就不会创建不必要的图表。