megengine.functional.nn.nvof# nvof(src, precision=1)[源代码]# NVIDIA 光流 SDK 的实现 参数: src (Tensor) – 输入 Tensor,要求 shape 为 (n, t, h, w, c4),dtype 为 unit8 precision (int) – 0:NV_OF_PERF_LEVEL_SLOW 1:NV_OF_PERF_LEVEL_MEDIUM 2:NV_OF_PERF_LEVEL_FAST. 返回: shape 为 (n, t-1, (h+out_grid_size-1)//out_grid_size, (w+out_grid_size-1)//out_grid_size, c2) (out_grad_size 默认为 4),dtype 为 int16 返回类型: output tensor with shape