()
| 704 | concatenated_shape[axis] = sum(tensor.shape[axis] for tensor in lazy_tensors) |
| 705 | |
| 706 | def load() -> UnquantizedTensor: |
| 707 | ndarrays = [load_unquantized(tensor) for tensor in lazy_tensors] |
| 708 | concatenated: NDArray = np.concatenate(ndarrays, axis=axis) |
| 709 | return UnquantizedTensor(concatenated) |
| 710 | description = 'concatenated[[' + '] | ['.join(lt.description for lt in lazy_tensors) + ']]' |
| 711 | return LazyTensor(load, concatenated_shape, lazy_tensors[0].data_type, description) |
| 712 | return {name: convert(name) for name in names} |
no test coverage detected