apollo::prediction::network::Flatten类 参考

network layer to flatten a matrix into vector. 更多...

#include <net_layer.h>

类 apollo::prediction::network::Flatten 继承关系图:

apollo::prediction::network::Flatten 的协作图:

Public 成员函数
bool	Load (const apollo::prediction::LayerParameter &layer_pb) override
	Load the dense layer parameter from a pb message
void	Run (const std::vector< Eigen::MatrixXf > &inputs, Eigen::MatrixXf *output) override
	Compute the layer output from inputs
Public 成员函数 继承自 apollo::prediction::network::Layer
	Layer ()=default
	Constructor
virtual	~Layer ()=default
	Destructor
virtual void	ResetState ()
	Reset the internal state of a layer such as LSTM, GRU
virtual void	SetState (const std::vector< Eigen::MatrixXf > &states)
	Set the internal state of a layer
virtual void	State (std::vector< Eigen::MatrixXf > *states) const
	Access to the internal state of a layer
std::string	Name () const
	Name of a layer
int	OrderNumber () const
	Order number of a layer in a network

详细描述

network layer to flatten a matrix into vector.

在文件 net_layer.h 第 395 行定义.

成员函数说明

Load()

bool apollo::prediction::network::Flatten::Load ( const apollo::prediction::LayerParameter &  layer_pb )

overridevirtual

Load the dense layer parameter from a pb message

参数

A	pb message contains the parameters

返回

True is loaded successively, otherwise False

重载 apollo::prediction::network::Layer .

在文件 net_layer.cc 第 487 行定义.

                                                 {
  if (!Layer::Load(layer_pb)) {
    AERROR << "Fail to Load the layer parameters!";
    return false;
  }
  return true;
}

Run()

void apollo::prediction::network::Flatten::Run ( const std::vector< Eigen::MatrixXf > &  inputs, Eigen::MatrixXf *  output  )

overridevirtual

Compute the layer output from inputs

参数

Inputs	to a network layer
Output	of a network layer will be returned

实现了 apollo::prediction::network::Layer.

在文件 net_layer.cc 第 495 行定义.

                                         {
  CHECK_EQ(inputs.size(), 1U);
  Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> inp(
      inputs[0]);
  inp.resize(1, inp.size());
  *output = inp;
}

---

该类的文档由以下文件生成:

• modules/prediction/network/net_layer.h
• modules/prediction/network/net_layer.cc