Python tensorrt.float32方法代碼示例,tensorrt.float32用法. ... x1: x2] = 0. mask = torch.from_numpy(mask) mask = mask.expand_as(img) img *= mask return img.
確定! 回上一頁