一、前言 本文以實現(xiàn)一個axis_abs的自定義算子為例介紹如何在tvm中添加新的relay算子傲绣，該算子實現(xiàn)的功能是以輸入的3維tensor取某一維度的指定切片取絕對值东帅。 二...
發(fā)簡信
IP屬地：浙江
一、前言 本文以實現(xiàn)一個axis_abs的自定義算子為例介紹如何在tvm中添加新的relay算子傲绣，該算子實現(xiàn)的功能是以輸入的3維tensor取某一維度的指定切片取絕對值东帅。 二...
