TensorFlow完整的運算符重載清單,強力收藏!

TensorFlow對常見數學運算符進行了重載,以使乘法、加法、減法及其他常見運算更加簡潔。如果運算符有一個或多個參數(操作對象)爲Tensor對象,則會有一個TensorFlow OP被調用,並被添加到數據流圖中。 例如,可按照下列方式輕鬆實現兩個張量的加法: # 假設a和b均爲Tensor對象,且形狀匹配,c=a+b 完整的張量重載運算符清單: 利用這些重載運算符可快速地對代碼進行整合,但卻無
相關文章
相關標籤/搜索