Shortcuts

mmedit.models.editors.liif.mlp_refiner

Module Contents

Classes

MLPRefiner

Multilayer perceptrons (MLPs), refiner used in LIIF.

class mmedit.models.editors.liif.mlp_refiner.MLPRefiner(in_dim, out_dim, hidden_list)[源代码]

Bases: mmengine.model.BaseModule

Multilayer perceptrons (MLPs), refiner used in LIIF.

参数
  • in_dim (int) – Input dimension.

  • out_dim (int) – Output dimension.

  • hidden_list (list[int]) – List of hidden dimensions.

forward(x)[源代码]

Forward function.

参数

x (Tensor) – The input of MLP.

返回

The output of MLP.

返回类型

Tensor

Read the Docs v: latest
Versions
master
latest
stable
zyh-doc-notfound-extend
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.