(self)
| 293 | |
| 294 | |
| 295 | def get_summarized_data(self): |
| 296 | dim = self.dim() |
| 297 | if dim == 0: |
| 298 | return self |
| 299 | if dim == 1: |
| 300 | if self.size(0) > 2 * PRINT_OPTS.edgeitems: |
| 301 | return flow.cat( |
| 302 | (self[: PRINT_OPTS.edgeitems], self[-PRINT_OPTS.edgeitems :]) |
| 303 | ) |
| 304 | else: |
| 305 | return self |
| 306 | if self.size(0) > 2 * PRINT_OPTS.edgeitems: |
| 307 | start = [self[i] for i in range(0, PRINT_OPTS.edgeitems)] |
| 308 | end = [ |
| 309 | self[i] for i in range(self.shape[0] - PRINT_OPTS.edgeitems, self.shape[0]) |
| 310 | ] |
| 311 | return flow.stack([get_summarized_data(x) for x in (start + end)]) |
| 312 | else: |
| 313 | return flow.stack([get_summarized_data(x) for x in self]) |
| 314 | |
| 315 | |
| 316 | def _format_tensor_on_cpu(tensor): |
no test coverage detected