b3bc509ad60a7ce5befbb50275a350d5959ea520,mmdnn/conversion/pytorch/pytorch_emitter.py,PytorchEmitter,emit_Embedding,#PytorchEmitter#Any#,362

Before Change

    def emit_Embedding(self, IR_node):
        raise NotImplementedError()
        ret = "{:<15} = Embedding(input_dim = {}, output_dim = {}, mask_zero = {})({})".format(
                IR_node.name,
                IR_node.IR_layer.attr["input_dim"].i,
                IR_node.IR_layer.attr["output_dim"].i,
                IR_node.IR_layer.attr["mask_zero"].b,
                IR_node.in_edges[0])

        return ret

After Change

    def emit_Embedding(self, IR_node):
        self.used_layers.add("Embedding")
        self.add_init(2, "self.{} = self.__embedding("{}", num_embeddings={}, embedding_dim={})".format(
            IR_node.variable_name,
            IR_node.name,
            IR_node.get_attr("input_dim"),   /ǘ-D
            IR_node.get_attr("output_dim")
            ))
        self.add_body(2, "{:<15} = self.{}({})".format(
            IR_node.variable_name,
            IR_node.variable_name,

Instances