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

Method add_axisaddremove

src/gopt/test/network.cpp:145–148  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

143}
144
145SymbolVar Network::add_axisaddremove(SymbolVar f) {
146 return opr::AxisAddRemove::make(
147 f, {{opr::AxisAddRemove::AxisDesc::Method::REMOVE, {0}}});
148}
149
150SymbolVar Network::add_subtensor(SymbolVar f) {
151 using AIdx = opr::indexing::AxisIndexer;

Callers 1

Calls 1

makeFunction · 0.50

Tested by

no test coverage detected