Get the specs for the input tensor of the network. Useful to prepare memory allocations. :return: Two items, the shape of the input tensor and its (numpy) datatype.
(self)
| 97 | assert len(self.allocations) > 0 |
| 98 | |
| 99 | def input_spec(self): |
| 100 | """ |
| 101 | Get the specs for the input tensor of the network. Useful to prepare memory allocations. |
| 102 | :return: Two items, the shape of the input tensor and its (numpy) datatype. |
| 103 | """ |
| 104 | return self.inputs[0]['shape'], self.inputs[0]['dtype'] |
| 105 | |
| 106 | def output_spec(self): |
| 107 | """ |