Network forwarding given input tensors Parameters ---------- name : str or None Name of the requested layer. Default None. index : int or None Index of the requested layer. Default None. Returns ------- layer : The
(self, name=None, index=None)
| 655 | raise Exception("all_drop is deprecated") |
| 656 | |
| 657 | def get_layer(self, name=None, index=None): |
| 658 | """Network forwarding given input tensors |
| 659 | |
| 660 | Parameters |
| 661 | ---------- |
| 662 | name : str or None |
| 663 | Name of the requested layer. Default None. |
| 664 | index : int or None |
| 665 | Index of the requested layer. Default None. |
| 666 | |
| 667 | Returns |
| 668 | ------- |
| 669 | layer : The requested layer |
| 670 | |
| 671 | Notes |
| 672 | ----- |
| 673 | Either a layer name or a layer index should be given. |
| 674 | |
| 675 | """ |
| 676 | if index is not None: |
| 677 | if len(self.all_layers) <= index: |
| 678 | raise ValueError( |
| 679 | 'model only has ' + str(len(self.all_layers)) + ' layers, but ' + str(index) + |
| 680 | '-th layer is requested.' |
| 681 | ) |
| 682 | else: |
| 683 | return self.all_layers[index] |
| 684 | elif name is not None: |
| 685 | for layer in self.all_layers: |
| 686 | if layer.name == name: |
| 687 | return layer |
| 688 | raise ValueError('Model has no layer named ' + name + '.') |
| 689 | else: |
| 690 | raise ValueError('Either a layer name or a layer index should be given.') |
| 691 | |
| 692 | def _construct_graph(self): |
| 693 | """construct computation graph for static model using LayerNode object""" |
no outgoing calls