close
機器之心報道
編輯:陳萍
TensorFlow 2.8.0 來了。

近日 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

arrow
arrow
    全站熱搜
    創作者介紹
    創作者 鑽石舞台 的頭像
    鑽石舞台

    鑽石舞台

    鑽石舞台 發表在 痞客邦 留言(0) 人氣()