cubic function used for calculate_weights_indices.
(x)
| 4 | |
| 5 | |
| 6 | def cubic(x): |
| 7 | """cubic function used for calculate_weights_indices.""" |
| 8 | absx = torch.abs(x) |
| 9 | absx2 = absx**2 |
| 10 | absx3 = absx**3 |
| 11 | return (1.5 * absx3 - 2.5 * absx2 + 1) * ( |
| 12 | (absx <= 1).type_as(absx)) + (-0.5 * absx3 + 2.5 * absx2 - 4 * absx + 2) * (((absx > 1) * |
| 13 | (absx <= 2)).type_as(absx)) |
| 14 | |
| 15 | |
| 16 | def calculate_weights_indices(in_length, out_length, scale, kernel, kernel_width, antialiasing): |
no outgoing calls
no test coverage detected