You can use the shape_array function on arrays to retrieve shapes as an MXNet managed array (i.e. symbol/ndarray) and use that to retrieve the num_categories as a scalar array through indexing
-- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/apache/incubator-mxnet/issues/19722#issuecomment-754630802
