提交 463650f1 编写于 作者: D dolphin8

fix softmax

上级 cf8f65b6
...@@ -33,17 +33,17 @@ __kernel void softmax(__read_only image2d_t input_image, ...@@ -33,17 +33,17 @@ __kernel void softmax(__read_only image2d_t input_image,
maxv = max(maxv, max(temp.x, max(temp.y, max(temp.z, temp.w)))); maxv = max(maxv, max(temp.x, max(temp.y, max(temp.z, temp.w))));
} }
half4 rsum = (half4)(0.0f);
half4 rsum = (half4)(0.0f);
for (int i = 0; i < group; ++i) { for (int i = 0; i < group; ++i) {
half4 r = read_imageh(input_image, sampler, (int2)(i, 0)); half4 r = read_imageh(input_image, sampler, (int2)(i, 0));
rsum += exp(r - maxv); rsum += convert_half4(exp(convert_float4(r - maxv)));
} }
float sum = rsum.x + rsum.y + rsum.z + rsum.w; float sum = rsum.x + rsum.y + rsum.z + rsum.w;
half4 rr = read_imageh(input_image, sampler, (int2)(out_w, out_nh)); half4 rr = read_imageh(input_image, sampler, (int2)(out_w, out_nh));
half4 result = exp(rr - maxv) / sum; half4 result = convert_half4(exp(convert_float4(rr - maxv)) / sum);
write_imageh(output_image, (int2)(out_w, out_nh), result); write_imageh(output_image, (int2)(out_w, out_nh), result);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册