(input_sizes, kernel_sizes, stride, padding)
| 88 | |
| 89 | |
| 90 | def cal_outsize(input_sizes, kernel_sizes, stride, padding): |
| 91 | stride_d, stride_h, stride_w = stride |
| 92 | padding_d, padding_h, padding_w = padding |
| 93 | dilation_d, dilation_h, dilation_w = 1, 1, 1 |
| 94 | |
| 95 | in_d = input_sizes[1] |
| 96 | in_h = input_sizes[2] |
| 97 | in_w = input_sizes[3] |
| 98 | in_channel = input_sizes[4] |
| 99 | |
| 100 | |
| 101 | kernel_d = kernel_sizes[2] |
| 102 | kernel_h = kernel_sizes[3] |
| 103 | kernel_w = kernel_sizes[4] |
| 104 | out_channels = kernel_sizes[0] |
| 105 | |
| 106 | out_d = calc_out_(in_d, padding_d, dilation_d, kernel_d, stride_d) |
| 107 | out_h = calc_out_(in_h, padding_h, dilation_h, kernel_h, stride_h) |
| 108 | out_w = calc_out_(in_w, padding_w, dilation_w, kernel_w, stride_w) |
| 109 | size = [input_sizes[0], out_d, out_h, out_w, out_channels] |
| 110 | return size |
| 111 | |
| 112 | |
| 113 |
no test coverage detected