MCPcopy Create free account
hub / github.com/dromara/Omega-AI / addAxisBack

Method addAxisBack

src/main/java/com/omega/engine/ad/op/TensorOP.java:61–65  ·  view source on GitHub ↗
(Tensor dx, Tensor dy,int N, int C, int H, int W, int axis)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

modulate_backMethod · 0.80
diffMethod · 0.80
diffMethod · 0.80
modulate_backMethod · 0.80
diffMethod · 0.80
diffMethod · 0.80
modulate_backMethod · 0.80
modulate_backMethod · 0.80
modulate_backMethod · 0.80
modulate_back2Method · 0.80
modulate_back2Method · 0.80
modulate_backMethod · 0.80

Calls 2

isHasGPUMethod · 0.80
add_axis_back_gpuMethod · 0.80

Tested by

no test coverage detected