Fork自 PaddlePaddle / Paddle
* Use BroadcastKernel and ReduceKernel to optimize expand and expand_grad. * Correct the axis when there is only 1 input in BroadcastKernel. * Add the calculate of output's shape.
拖放文件到此处或点击上传