pytorch_emitter.py 文件源码

python
阅读 28 收藏 0 点赞 0 评论 0
项目：MMdnn 作者: Microsoft 项目源码 文件源码 
def emit_Squeeze(self, IR_node):
        self.add_body(2, "{:<15} = torch.squeeze({})".format(
            IR_node.variable_name, self.parent_variable_name(IR_node)
        ))
您未登录，请先登录后再评论
评论列表
文章目录

提
问题

写
面经

写
文章

微信
公众号

扫码关注公众号