MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / add_axis

Method add_axis

src/core/impl/graph/symbol_var.cpp:41–44  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

39}
40
41SymbolVar SymbolVar::add_axis(size_t idx) const {
42 return opr::AxisAddRemove::make(
43 *this, {opr::AxisAddRemove::AxisDesc::make_add(idx)});
44}
45
46Maybe<DTypeScalar> SymbolVar::as_immutable_scalar() const {
47 using IT = static_infer::InferType;

Callers 1

run_graphFunction · 0.80

Calls 1

makeFunction · 0.70

Tested by 1

run_graphFunction · 0.64