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

Method add_subtensor

src/gopt/test/network.cpp:150–154  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

148}
149
150SymbolVar Network::add_subtensor(SymbolVar f) {
151 using AIdx = opr::indexing::AxisIndexer;
152 return opr::Subtensor::make(
153 f, {AIdx::make_interval(0, f.make_scalar(0), None, None)});
154}
155
156SymbolVar Network::add_reshape(SymbolVar f) {
157 auto shp = opr::GetVarShape::make(f);

Callers 1

Calls 2

makeFunction · 0.50
make_scalarMethod · 0.45

Tested by

no test coverage detected