在Tensorflow中檢查廣播兼容性 - C++ API

Question

我正在TensorFlow中實施基於元素的操作。許多TensorFlow操作，例如添加，支持廣播（from numpy）。如果遵守以下規則，則廣播是可能的：

當在兩個張量上操作時，它們的形狀應該按照元素進行比較。該過程從尾部維度開始，並向前推進。兩個維度在相等時兼容，或者其中一個維度是1.如果這些條件不滿足，則拋出異常，表明張量具有不兼容的形狀。結果張量的大小是輸入數組每個維度的最大大小。

不TensorFlow C++ API提供用於比較兼容性2張量任何方法？或者，這是最快的方法嗎？

Answer 1

TensorFlow中的所有基於元素的二進制操作的內核實現來自BinaryOpShared類，它通過幫助類BinaryOpState進行兼容性檢查。也許，你可以簡單地從BinaryOpShared派生你的內核類，並免費獲得兼容性檢查。