print('Shape of X :',X.size()) print('Shape of Y :',Y.size()) print('Shape of Z :',z.size())