irtk.connectors.nvdiffrast_connector.avg_pool_nhwc

irtk.connectors.nvdiffrast_connector.avg_pool_nhwc(x: torch.Tensor, size) torch.Tensor