MCPcopy Create free account
hub / github.com/apache/singa / Axis

Method Axis

src/model/layer/flatten.h:45–45  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43 int flag, const Tensor &grad) override;
44
45 int Axis() const { return axis_; }
46 const Shape input_shape() const { return input_shape_; }
47
48 protected:

Callers 1

TESTFunction · 0.80

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.64