當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python mxnet.symbol.op.batch_dot用法及代碼示例


用法:

mxnet.symbol.op.batch_dot(lhs=None, rhs=None, transpose_a=_Null, transpose_b=_Null, forward_stype=_Null, name=None, attr=None, out=None, **kwargs)

參數

  • lhs(Symbol) - 第一個輸入
  • rhs(Symbol) - 第二個輸入
  • transpose_a(boolean, optional, default=0) - 如果為真,則轉置點之前的第一個輸入。
  • transpose_b(boolean, optional, default=0) - 如果為真,則轉置點之前的第二個輸入。
  • forward_stype({None, 'csr', 'default', 'row_sparse'},optional, default='None') - 用戶給定的前向輸出的所需存儲類型,如果輸入存儲類型和此提示的組合與任何已實現的存儲類型不匹配,點運算符將執行回退操作並仍產生所需存儲類型的輸出。
  • name(string, optional.) - 結果符號的名稱。

返回

結果符號。

返回類型

Symbol

批量點積。

xy是批量數據時,batch_dot用於計算xy的點積,即N-D(N >= 3)個數組,形狀為(B0, …, B_i, :, :)

例如,給定形狀為 (B_0, …, B_i, N, M)x 和形狀為 (B_0, …, B_i, M, K)y ,結果數組的形狀為 (B_0, …, B_i, N, K) ,其計算公式為:

batch_dot(x,y)[b_0, ..., b_i, :, :] = dot(x[b_0, ..., b_i, :, :], y[b_0, ..., b_i, :, :])

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.symbol.op.batch_dot。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。