See also: https://pytorch.org/docs/stable/generated/torch.nn.functional.grid_sample.html interpolation mode of `torch.nn.functional.grid_sample` Note: (documentation from `torch.nn.functional.grid_sample`) `mode='bicubic'` supports only 4-D input. When `mode='b
| 125 | |
| 126 | |
| 127 | class GridSampleMode(StrEnum): |
| 128 | """ |
| 129 | See also: https://pytorch.org/docs/stable/generated/torch.nn.functional.grid_sample.html |
| 130 | |
| 131 | interpolation mode of `torch.nn.functional.grid_sample` |
| 132 | |
| 133 | Note: |
| 134 | (documentation from `torch.nn.functional.grid_sample`) |
| 135 | `mode='bicubic'` supports only 4-D input. |
| 136 | When `mode='bilinear'` and the input is 5-D, the interpolation mode used internally will actually be trilinear. |
| 137 | However, when the input is 4-D, the interpolation mode will legitimately be bilinear. |
| 138 | """ |
| 139 | |
| 140 | NEAREST = "nearest" |
| 141 | BILINEAR = "bilinear" |
| 142 | BICUBIC = "bicubic" |
| 143 | |
| 144 | |
| 145 | class SplineMode(StrEnum): |
no outgoing calls
no test coverage detected
searching dependent graphs…