如何为 PyTorch 接入自定义 C++ 或 CUDA 算子？

知识专题：训练框架、调参与排障题目类型：操作实践技术难度：进阶

收录题集：4 份

暂无参考答案

未标记

询问 AI

选择 AI 平台

Kimi 千问 ChatGPT

自动带入当前题目

本题目录

收录题集

4 份

AI系统与训练推理工程

阿东玩 AI · AgentGuide · 1 条原题

查看题集 →

- 【TensorFlow插件与PyTorch扩展对比】TensorFlow插件机制为什么让人觉得麻烦，和PyTorch的extension比起来差别在哪？「阿里巴巴」

自定义算子实现

Rocky Ding · AIGC-Interview-Book · 1 条原题

查看题集 →

- 介绍一下模型部署领域自定义算子的概念

模型部署精华 · ONNX 与 TensorRT

Rocky Ding · AIGC-Interview-Book · 1 条原题

查看题集 →

- TensorRT部署中遇到不支持算子或插件算子时如何处理？

PyTorch 实现与追问

Alireza Dirafzoon · AIMLInterviews · 1 条原题

查看题集 →

题目来源: Devinterview 其中第 85—146 题的题目索引来自 Devinterview，答案由 AIMLInterviews 重新撰写；原文的 50 题补充部分在此收录为 62 道，含拆分追问。

- How do you create a custom C++ or CUDA operation?