wrote the Frobenius norm for the PSGD; I get Arrout of of Index. some tiny bug