Class Net
Defined in File net.h
Inheritance Relationships
Base Types
public mindspore::NetBase
public std::enable_shared_from_this< Net >
Class Documentation
-
class Net : public mindspore::NetBase, public std::enable_shared_from_this<Net>
Public Functions
-
virtual std::vector<Expr*> construct(const std::vector<Expr*> &inputs)
Define the relation between network inputs and outputs.
- 参数
inputs – [in] expression vector
- 返回
expression vector
-
std::shared_ptr<NodeSet> trainable_params()
Find the trainable params for the trained network.
- 返回
NodeSet for all trainable nodes
-
const std::vector<int> InputShape(int idx)
Input shape.
- 参数
idx – [in] input index
- 返回
Specific input shape vector
-
const std::vector<int> OutputShape(int idx)
Output shape.
- 参数
idx – [in] Output index
- 返回
Specific output shape vector
-
virtual std::vector<Expr*> construct(const std::vector<Expr*> &inputs)