(
rebuild_compressed_indices_func,
rebuild_compressed_indices_args,
rebuild_plain_indices_func,
rebuild_plain_indices_args,
rebuild_values_func,
rebuild_values_args,
shape,
layout,
)
| 459 | |
| 460 | |
| 461 | def rebuild_sparse_compressed_tensor( |
| 462 | rebuild_compressed_indices_func, |
| 463 | rebuild_compressed_indices_args, |
| 464 | rebuild_plain_indices_func, |
| 465 | rebuild_plain_indices_args, |
| 466 | rebuild_values_func, |
| 467 | rebuild_values_args, |
| 468 | shape, |
| 469 | layout, |
| 470 | ): |
| 471 | compressed_indices = rebuild_compressed_indices_func( |
| 472 | *rebuild_compressed_indices_args |
| 473 | ) |
| 474 | plain_indices = rebuild_plain_indices_func(*rebuild_plain_indices_args) |
| 475 | values = rebuild_values_func(*rebuild_values_args) |
| 476 | return torch.sparse_compressed_tensor( |
| 477 | compressed_indices, plain_indices, values, shape, layout=layout |
| 478 | ) |
| 479 | |
| 480 | |
| 481 | def reduce_sparse_tensor(sparse): |
nothing calls this directly
no outgoing calls
no test coverage detected