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

用法:
mxnet.ndarray.op.batch_dot(lhs=None, rhs=None, transpose_a=_Null, transpose_b=_Null, forward_stype=_Null, out=None, name=None, **kwargs)

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

out：- 此函數的輸出。

NDArray 或 NDArray 列表

批量點積。

當x和y是批量數據時，batch_dot用於計算x和y的點積，即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.ndarray.op.batch_dot。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。