Method GetVarShape

src/opr/impl/tensor_manip.cpp:68–78

MGB_DYN_TYPE_OBJ_FINAL_IMPL(GetVarShape);
GetVarShape::GetVarShape(
        const VarNodeArrayView& inp, Param axis, const OperatorNodeConfig& config)
        : Super(inp.at(0)->owner_graph(), config, "shape_of", inp), m_axis{axis} {
    m_src_shapes.resize(inp.size());
    for (auto i : inp)
        add_input({i});
    add_input({}, AddInputSortType::ALL);
    add_output(None)->dtype(dtype::Int32());
    add_equivalence_component<PODHash<Param>>(&m_axis);
    mgb_assert(abs(m_axis.axis) <= m_axis.MAX_NDIM);
}

void GetVarShape::update_cached_shape() {
    TensorShape ishp;