--- a +++ b/Graph_Conversion.py @@ -0,0 +1,21 @@ +from torch.cuda import LongTensor +from numpy import ndarray, array, int32, append, flip + +def Convert_To_Image(tensor: LongTensor, adj_count: int) -> ndarray: + r""" + Arguments: + tensor (torch.cuda.LongTensor): Graph as tensor. + + Returns: + out (numpy.ndarray): Converted image. + """ + tensor = tensor.cpu() + out = array([tensor[0:512]], dtype = int32) + + for i in range(1, 512): + if i % 2 == 1: + out = append(out, flip([tensor[i*adj_count:adj_count+i*adj_count]]), axis = 0) + else: + out = append(out, [tensor[i*adj_count:adj_count+i*adj_count]], axis = 0) + + return out \ No newline at end of file