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