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