簡體   English   中英

如何在批處理數據上使用點生產?

[英]how to use dot production on batch data?

我正在嘗試使用tanh(dot(x,y)) ; x和y是我的RNN的批處理數據。

x,y的形狀為(n_batch,n_length,n_dim)像(2,3,4); 具有3個序列的2個樣本,每個樣本為4維。

我想對最后一個尺寸進行內部或點生產。 然后tanh(dot(x,y))的形狀應為(n_batch,n_length)=(2,3)

我應該使用哪個功能?

這個表達式應該可以解決這個問題:

theano.tensor.tanh((x * y).sum(2))

點積是通過逐元素相乘,然后對最后一個維度求和來“手動”計算的。

暫無
暫無

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

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