什么是tf.TensorSpec()以及它的用途是什么?
什么是 TensorSpec?(张量流 2.0)
数据挖掘
神经网络
深度学习
张量流
python-3.x
2022-03-10 00:59:06
1个回答
tf.function 主要使用 TensorSpec 来指定输入签名。tf.function 将为不同的输入形状和数据类型创建一个图,但您的函数图可能与不同的形状兼容。作为性能优化,您可以选择提供一个签名,这样就不会创建不必要的图表。