Class Net

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