thanks alot…
i have another question…
if i want to remove row and column in one hidden layer after train (just for test phase) and repeat  the  above row and column to this row, how i work and coding in pytorch…
thanks for helping…