hmmm… I dont know for sure this line in tensorflow. Its better if you explain the meaning of that line first, so another people that does not have experience in tensorflow can give you solution.
But anyway, for concate a tensor or image you can use
torch.cat([image_tensor1, image_tensor2])