|
|
|
@ -19,6 +19,7 @@ from ..cell import Cell
|
|
|
|
|
|
|
|
|
|
__all__ = ['SequentialCell', 'CellList']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _valid_index(cell_num, index):
|
|
|
|
|
if not isinstance(index, int):
|
|
|
|
|
raise TypeError("Index {} is not int type")
|
|
|
|
@ -145,6 +146,12 @@ class SequentialCell(Cell):
|
|
|
|
|
for cell in self._cells.values():
|
|
|
|
|
cell.set_grad(flag)
|
|
|
|
|
|
|
|
|
|
def append(self, cell):
|
|
|
|
|
"""Appends a given cell to the end of the list."""
|
|
|
|
|
if _valid_cell(cell):
|
|
|
|
|
self._cells[str(len(self))] = cell
|
|
|
|
|
return self
|
|
|
|
|
|
|
|
|
|
def construct(self, input_data):
|
|
|
|
|
for cell in self.cell_list:
|
|
|
|
|
input_data = cell(input_data)
|
|
|
|
|