Table of Contents
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