TensorFlow2自定义算子

• 2022年07月31日

仓库地址：https://github.com/tensorflow/custom-op 将仓库拉到本地，假设自定义算子名字是inno，复制zero_out文件夹重命名为tensorflow_inno，功能是（1）保留input的第一个数字；（2）input其他位置上的元素都变为原来2倍。…