a277fd89980f7a4647d0a17548aba7df37a5aefb,pyprob/nn/inference_network_feedforward.py,InferenceNetworkFeedForward,infer_trace_init,#InferenceNetworkFeedForward#Any#,159

Before Change

    def infer_trace_init(self, observe=None):
        self._infer_observe = observe
        self._infer_observe_embedding = self._embed_observe({name: util.to_tensor(v) for name, v in observe.items()})

    def infer_trace_step(self, variable, previous_variable=None):
        success = True
        address = variable.address

After Change

        self._infer_observe = observe
        embedding = []
        for name, layer in self._layer_observe_embedding.items():
            value = util.to_tensor(observe[name]).view(1, -1)
            embedding.append(layer(value))
        embedding = torch.cat(embedding, dim=1)
        self._infer_observe_embedding = self._layer_observe_embedding_final(embedding)

Instances