提交 91432cb5 编写于 作者: G guo ran 提交者: Jinhui Yuan

fix Div function (#1212)

上级 87db37ed
......@@ -90,7 +90,7 @@ struct SoftmaxKernelUtil<DeviceType::kCPU, T> {
static void Div(DeviceCtx* ctx, const int64_t n, const int64_t w, T* matrix, const T* vector) {
for (int64_t i = 0; i < n; ++i) {
KernelUtil<DeviceType::kCPU, T>::Div(ctx, n, matrix + i * w, vector + i);
KernelUtil<DeviceType::kCPU, T>::Div(ctx, w, matrix + i * w, vector + i);
}
}
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册