mindspore.nn.CellList

class mindspore.nn.CellList(*args, **kwargs)

Holds Cells in a list. For more details about Cell, please refer to Cell.

CellList can be used like a regular Python list, the Cells it contains have been initialized and the types of Cells it contains can not be CellDict. Unlike the SequentialCell, the cells in CellList are not connected.

Parameters

args (list, optional) – List of subclass of Cell.

Supported Platforms:

Ascend GPU CPU

Examples

>>> 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.]]

append(cell)

Appends a given Cell to the end of the list.

Parameters

cell (Cell) – The subcell to be appended.

extend(cells)

Appends Cells from a Python iterable to the end of the list.

Parameters

cells (list) – The Cells to be extended, the types of Cells can not be CellDict.

Raises

TypeError – If the argument cells are not a list of Cells.

insert(index, cell)

Inserts a given Cell before a given index in the list.

Parameters

• index (int) – The Insert index in the CellList.

• cell (Cell) – The Cell to be inserted.