近日 TensorFlow 官方發布了 2.8.0 正式版,距離上次 2.7 版本的更新過去沒多久,新版本提供了更多的 bug 修復和功能改進,此外新版本還針對漏洞發布了補丁。
對於 TensorFlow 2.8.0 的上線,網友也紛紛感嘆,這次的 Bug 修復也太棒了!
不過也有網友已經迫不及待的期待 TensorFlow 3.0 的到來,並表示 TensorFlow 1 到 TensorFlow 2 是一個巨大的飛躍,將來 2 到 3 又會有哪些亮眼的表現呢?
眾所周知,TensorFlow 由谷歌開發,最初版本可追溯到 2015 年開源的 TensorFlow0.1,之後發展穩定,擁有強大的用戶群體,成為最受歡迎的深度學習框架。但是用戶在使用時,也暴露了 TensorFlow 缺點,例如 API 穩定性不足、靜態計算圖編程複雜等缺陷。因此在 TensorFlow2.0 版本,谷歌將 Keras 納入進來,成為 tf.keras,到目前為止,TensorFlow 已經更新到 2.8 版本,那麼新版本有哪些重要的改進呢?
TensorFlow 2.8.0 主要功能和改進
在 tf.lite 中,增加了 TFLite 內置 op 支持以下功能:
tf.raw_ops.Bucketize op 可在 CPU 上操作;
tf.where op 可用於數據類型 tf.int32、tf.uint32、tf.int8、tf.uint8、tf.int64;
tf.random.normal op 用於在 CPU 上輸出數據類型 tf.float32;
tf.random.uniform op 用於在 CPU 上輸出數據類型 tf.float32;
f.random.categorical op 用於在 CPU 上的輸出數據類型 tf.int64。
tensorflow.experimental.tensorrt:
Conversion_params 在 TrtGraphConverterV2 中被棄用,現在可以支持參數 max_workspace_size_bytes、precision_mode、minimum_segment_size、maximum_cached_engines、use_calibration 和 allow_build_at_runtime;
在 TrtGraphConverterV2 中的 .save () 函數中添加了一個名為 save_gpu_specific_engines 的新參數。當為 False 時,.save () 函數不會保存任何已構建的 TRT 引擎;如果為 True(默認),則保留原始行為;
TrtGraphConverterV2 提供了一個名為 .summary () 的新 API。它顯示了每個 TRTEngineOp 及其輸入和輸出的形狀和 dtype,並提供了詳細版本摘要。
tf.tpu.experimental.embedding:
tf.tpu.experimental.embedding.FeatureConfig 增加了一個額外的參數 output_shape,它可以指定特徵輸出激活的形狀;
tf.tpu.experimental.embedding.TPUEmbedding 現在具有與 tf.tpu.experimental.embedding.serving_embedding_lookup 相同的功能,它可以使用任意等級密集和稀疏的張量。對於不規則張量,儘管輸入張量仍然是 2 級,但現在可以通過在特徵配置中指定輸出形狀或通過 build 方法來激活 2 級或更高級別。
添加 tf.config.experimental.enable_op_determinism ,這使得 TensorFlow ops 以性能為代價可以確定性地運行。替換 TF_DETERMINISTIC_OPS 環境變量。
(自 TF 2.7 起)向 TensorFlow Profiler 添加 PluggableDevice 支持。
Bug 修復和其他改進
tf.data:
如果用戶未禁用,現在優化 parallel_batch 現在成為默認值,這樣可以並行複製批處理元素;
添加了 TensorSliceDataset,用於識別和處理文件輸入。
tf.lite:
為 Java API 的序列化添加 GPU 委託支持,當 OpenCL 可用時,這將初始化時間提高了 90%;
棄用 Interpreter::SetNumThreads,支持 InterpreterBuilder::SetNumThreads。
tf.keras
tf.random.Generator 用於 keras 初始化和所有的 RNG 代碼;
TextVectorization 增加了額外的 standardize 和 split 模式:standardize="lower" 轉化為小寫字母輸入;standardize="string_punctuation" 刪除所有標點符號;Split ="character" 將對每個 unicode 字符進行拆分。
增加 GPU 實現:
(自 2.7 版本開始) tf.math.segment_mean
(自 2.7 版本開始) tf.math.segment_prod
(自 2.7 版本開始) tf.math.segment_sum
TensorFlow 已在適用於 GPU 和 CPU 的 Windows Subsystem for Linux 2(又名 WSL 2)上得到驗證。
此外,TensorFlow 2.8.0 在安全方面進行了一些修正,包括修正了執行卷積運算時浮點數被 0 除的問題:CVE-2022-21725;修正了 Dequantize 形狀推斷中的整數溢出問題:CVE-2022-21727;修正了 ConcatV2 形狀推斷中的類型混淆問題:CVE-2022-21731 等。
更多內容,請參考:https://github.com/tensorflow/tensorflow/releases/tag/v2.8.0?linkId=8031153
©THE END
轉載請聯繫本公眾號獲得授權
投稿或尋求報道:content@jiqizhixin.com