Indicates a key can have a variable shape after collation
| 99 | |
| 100 | @dataclasses.dataclass |
| 101 | class VariablePaddingSpec: |
| 102 | """Indicates a key can have a variable shape after collation""" |
| 103 | |
| 104 | shape: Tuple[int] |
| 105 | """Shape of padding tensor if key is not in batch and padding is required""" |
| 106 | |
| 107 | dtype: np.dtype |
| 108 | """dtype of padding tensor if the key is not in batch and padding is required""" |
| 109 | |
| 110 | |
| 111 | def batch_pixels_to_patches(array, patch_size): |
no outgoing calls
no test coverage detected