体验新版 GitCode,发现更多精彩内容 >>
* 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.
拖放文件到此处或点击上传