def initialize_image_embedding()

in datasets.py [0:0]


    def initialize_image_embedding(self):
        w, h, c = self.embedding_sizes
        embedding = []
        for i in range(w):
            for j in range(h):
                for k in range(c):
                    embedding.append([i, j, k])
        self.x_emb = np.array(embedding).T.reshape([1, 3, self.ctx])