dmlc/mshadow

Broadcasting along multiple axis

sxjscience opened this issue · 1 comments

Sometimes we need to broadcast along some axis given in an std::vector -> broadcast_with_multi_axis. I'm not sure how to code this concisely for Tensor with arbitrary ndim, since we may not know the number of broadcasted axises in prior. One idea I have is to constrain the maximum axis of the broadcast_with_multi_axis to a large number like 5. Is this solution acceptable?

I find this may not be needed. Close this issue.