wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-03 18:23:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bug fixes to VerticalModel

Browse Source
This commit is contained in:
rowanc1
2014-02-05 22:09:24 -08:00
parent 72c93cd48b
commit ef6ad1d083
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Model.py
+6 -10
View File
@@ -145,7 +145,9 @@ class Vertical1DModel(BaseModel):
@property
def nP(self):
"""The number of cells in the
"""Number of model properties.
The number of cells in the
last dimension of the mesh."""
return self.mesh.nCv[self.mesh.dim-1]
@@ -155,7 +157,7 @@ class Vertical1DModel(BaseModel):
:rtype: numpy.array
:return: transformed model
"""
repNum = self.mesh.nCv[:self.mesh.dim-2].prod()
repNum = self.mesh.nCv[:self.mesh.dim-1].prod()
return Utils.mkvc(m).repeat(repNum)
def transformDeriv(self, m):
@@ -164,15 +166,9 @@ class Vertical1DModel(BaseModel):
:rtype: scipy.csr_matrix
:return: derivative of transformed model
"""
repNum = self.mesh.nCv[:self.mesh.dim-2].prod()
repNum = self.mesh.nCv[:self.mesh.dim-1].prod()
repVec = sp.csr_matrix(
(np.ones(repNum),
(range(repNum), np.zeros(repNum))
), shape=(repNum, 1))
return sp.kron(repVec, sp.identity(self.nP))
if __name__ == '__main__':
from SimPEG import *
mesh = Mesh.TensorMesh([10,8])
model = BaseModel(mesh)
model.test()
return sp.kron(sp.identity(self.nP), repVec)