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

Class Dimshuffle

imperative/python/megengine/utils/network_node.py:690–698  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

688
689
690class Dimshuffle(OpNode):
691 type = "Dimshuffle"
692 opdef = builtin.Dimshuffle
693
694 @classmethod
695 def load(cls, opr):
696 obj = super(Dimshuffle, cls).load(opr)
697 del obj.params["ndim"]
698 return obj
699
700
701class Reshape(OpNode):

Callers 1

layerPixelShuffleFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected