mindspore.nn.CellList
- class mindspore.nn.CellList(*args, **kwargs)[源代码]
构造Cell列表。关于Cell的介绍,可参考 Cell。
CellList可以像普通Python列表一样使用,其包含的Cell均已初始化。
- 参数:
args (list,可选) - Cell列表。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.nn as nn >>> >>> conv = nn.Conv2d(100, 20, 3) >>> bn = nn.BatchNorm2d(20) >>> relu = nn.ReLU() >>> cell_ls = nn.CellList([bn]) >>> cell_ls.insert(0, conv) >>> cell_ls.append(relu) >>> cell_ls.extend([relu, relu])