diff --git a/simpegEM/TDEM/SurveyTDEM.py b/simpegEM/TDEM/SurveyTDEM.py index 845cb515..c9a67856 100644 --- a/simpegEM/TDEM/SurveyTDEM.py +++ b/simpegEM/TDEM/SurveyTDEM.py @@ -146,14 +146,11 @@ class SurveyTDEM(Survey.BaseSurvey): for tx in self.txList: for rx in tx.rxList: Ptv = rx.projectFieldsDeriv(tx, self.mesh, self.prob.timeMesh, u, v, adjoint=True) + Ptv = Ptv.reshape((-1, self.prob.timeMesh.nN), order='F') if rx.projField not in f: # first time we are projecting - Ptv = Ptv.reshape((-1, 1, self.prob.timeMesh.nN), order='F') f[tx, rx.projField, :] = Ptv - else: - Ptv = Ptv.reshape((-1, self.prob.timeMesh.nN), order='F') - addedPtv = f[tx, rx.projField, :] + Ptv - addedPtv = addedPtv.reshape((-1, 1, self.prob.timeMesh.nN), order='F') - f[tx, rx.projField, :] = addedPtv + else: # there are already fields, so let's add to them! + f[tx, rx.projField, :] += Ptv return f