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)
| 83 | assert len(self.allocations) > 0 |
| 84 | |
| 85 | def input_spec(self): |
| 86 | """ |
| 87 | Get the specs for the input tensor of the network. Useful to prepare memory allocations. |
| 88 | :return: Two items, the shape of the input tensor and its (numpy) datatype. |
| 89 | """ |
| 90 | return self.inputs[0]["shape"], self.inputs[0]["dtype"] |
| 91 | |
| 92 | def output_spec(self): |
| 93 | """ |