(kernel_size: Sequence[int] | int, stride: Sequence[int] | int)
| 44 | |
| 45 | |
| 46 | def stride_minus_kernel_padding(kernel_size: Sequence[int] | int, stride: Sequence[int] | int) -> tuple[int, ...] | int: |
| 47 | kernel_size_np = np.atleast_1d(kernel_size) |
| 48 | stride_np = np.atleast_1d(stride) |
| 49 | |
| 50 | out_padding_np = stride_np - kernel_size_np |
| 51 | out_padding = tuple(int(p) for p in out_padding_np) |
| 52 | |
| 53 | return out_padding if len(out_padding) > 1 else out_padding[0] |
| 54 | |
| 55 | |
| 56 | def calculate_out_shape( |
no outgoing calls
no test coverage detected
searching dependent graphs…