Torch Conat Matrix
third_tensor = torch.cat((first_tensor, second_tensor), 0) # keep column width append in rows
third_tensor = torch.cat((first_tensor, second_tensor), 1) # keep row height and append in columns
mrjakobdk