簡體   English   中英

當你使用 tensorflow.keras.layers.Add 不同的形狀時會發生什么?

[英]What happens when you use tensorflow.keras.layers.Add on different shapes?

所以我只是根據文檔意識到,tensorflow keras 的添加層希望添加的張量之間具有相同的形狀(我知道,但錯誤地改變了一個的形狀)。 但是它沒有拋出錯誤,而是做了一些意想不到的事情。 有誰知道它的作用?

根據我的經驗,它復制並擴展較小形狀的張量以匹配較大形狀的張量(如果可能)並將它們相加以給出結果。 所以如果你這樣做

x1 = tf.zeros((32, 32, 3))
x2= tf.zeros((32, 32, 1))
temp = tf.keras.layers.add([x1, x2])
print(tf.shape(temp))

它會打印 (32, 32, 3)

也許不是非常重要的信息,但想防止人們通過使用添加層而不是獲得他們期望的形狀來犯和我一樣的錯誤!

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM