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])