模型开发接口
推理部署接口
API 相关说明
NVIDIA 光流 SDK 的实现
src (Tensor) – 输入 Tensor,要求 shape 为 (n, t, h, w, c4),dtype 为 unit8
Tensor
precision (int) – 0:NV_OF_PERF_LEVEL_SLOW 1:NV_OF_PERF_LEVEL_MEDIUM 2:NV_OF_PERF_LEVEL_FAST.
int
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
(n, t-1, (h+out_grid_size-1)//out_grid_size, (w+out_grid_size-1)//out_grid_size, c2)
output tensor with shape
上一页
megengine.functional.nn.correlation
下一页
megengine.functional.quantized.conv_bias_activation