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