Share boundaries between neighboring blocks Parameters ---------- x: da.Array A dask array depth: dict The size of the shared boundary per axis boundary: dict The boundary condition on each axis. Options are 'reflect', 'periodic', 'nearest', 'non
(x, depth, boundary, *, allow_rechunk=True)
| 389 | |
| 390 | |
| 391 | def overlap(x, depth, boundary, *, allow_rechunk=True): |
| 392 | """Share boundaries between neighboring blocks |
| 393 | |
| 394 | Parameters |
| 395 | ---------- |
| 396 | |
| 397 | x: da.Array |
| 398 | A dask array |
| 399 | depth: dict |
| 400 | The size of the shared boundary per axis |
| 401 | boundary: dict |
| 402 | The boundary condition on each axis. Options are 'reflect', 'periodic', |
| 403 | 'nearest', 'none', or an array value. Such a value will fill the |
| 404 | boundary with that value. |
| 405 | allow_rechunk: bool, keyword only |
| 406 | Allows rechunking, otherwise chunk sizes need to match and core |
| 407 | dimensions are to consist only of one chunk. |
| 408 | |
| 409 | The depth input informs how many cells to overlap between neighboring |
| 410 | blocks ``{0: 2, 2: 5}`` means share two cells in 0 axis, 5 cells in 2 axis. |
| 411 | Axes missing from this input will not be overlapped. |
| 412 | |
| 413 | Any axis containing chunks smaller than depth will be rechunked if |
| 414 | possible, provided the keyword ``allow_rechunk`` is True (recommended). |
| 415 | |
| 416 | Examples |
| 417 | -------- |
| 418 | >>> import numpy as np |
| 419 | >>> import dask.array as da |
| 420 | |
| 421 | >>> x = np.arange(64).reshape((8, 8)) |
| 422 | >>> d = da.from_array(x, chunks=(4, 4)) |
| 423 | >>> d.chunks |
| 424 | ((4, 4), (4, 4)) |
| 425 | |
| 426 | >>> g = da.overlap.overlap(d, depth={0: 2, 1: 1}, |
| 427 | ... boundary={0: 100, 1: 'reflect'}) |
| 428 | >>> g.chunks |
| 429 | ((8, 8), (6, 6)) |
| 430 | |
| 431 | >>> np.array(g) |
| 432 | array([[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100], |
| 433 | [100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100], |
| 434 | [ 0, 0, 1, 2, 3, 4, 3, 4, 5, 6, 7, 7], |
| 435 | [ 8, 8, 9, 10, 11, 12, 11, 12, 13, 14, 15, 15], |
| 436 | [ 16, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 23], |
| 437 | [ 24, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 31], |
| 438 | [ 32, 32, 33, 34, 35, 36, 35, 36, 37, 38, 39, 39], |
| 439 | [ 40, 40, 41, 42, 43, 44, 43, 44, 45, 46, 47, 47], |
| 440 | [ 16, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 23], |
| 441 | [ 24, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 31], |
| 442 | [ 32, 32, 33, 34, 35, 36, 35, 36, 37, 38, 39, 39], |
| 443 | [ 40, 40, 41, 42, 43, 44, 43, 44, 45, 46, 47, 47], |
| 444 | [ 48, 48, 49, 50, 51, 52, 51, 52, 53, 54, 55, 55], |
| 445 | [ 56, 56, 57, 58, 59, 60, 59, 60, 61, 62, 63, 63], |
| 446 | [100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100], |
| 447 | [100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100]]) |
| 448 | """ |