apache / mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
https://mxnet.apache.org
Apache License 2.0
20.77k stars 6.79k forks source link

TODO list for supporting dynamic shape #12732

Open zheng-da opened 5 years ago

zheng-da commented 5 years ago

MXNet doesn't support operators that don't allow static shape inference, such as unique and boolean_index. One possible solution is proposed in https://cwiki.apache.org/confluence/display/MXNET/Dynamic+shape

This task can be broken into multiple steps:

vrakesh commented 5 years ago

@zheng-da thanks for the detailed proposal on supporting dynamic shapes

@mxnet-label-bot [Feature request]

vrakesh commented 5 years ago

@mxnet-label-bot [Operator]

chinakook commented 5 years ago

I think that a good solution to support pseudo-dynamic (in most cases it's enough) shape is to support infer_shape_partial (through defer shape inference) in HybridBlock.