set_theano_shared_pos_magnetics

简介

该函数是Gempy中的InterpolatorModel类的一个方法，用于根据位置和磁性数据设置Theano共享变量。

参数

• positions: numpy.array
  • 一个(m,3)大小的数组，表示沉积物/岩石的位置（x，y，z）坐标。
• magnetization: numpy.array
  • 一个(m,3)大小的数组，表示沉积物/岩石在(x，y，z)方向上的磁性。
• dtype: str or None, default 'float32'
  • 共享变量的数据类型。如果为None，则使用默认值'float32'。
• verbose: bool, default False
  • 是否显示信息。

返回值

• 无返回值。

示例

import gempy as gp
import numpy as np

geo_model = gp.create_model('test')
gp.init_data(geo_model, extent=[0,10,0,10,0,10], resolution=[2,2,2])
gp.map_stack_to_surfaces(geo_model, {"rock": "rock_unit"})
gp.set_interpolator(geo_model,
                    compile_theano=True,
                    theano_optimizer='fast_run')

geo_model.interpolator.set_theano_shared_pos_magnetics(np.array([[1,1,1], [2,2,2]]),
                                                       np.array([[3,3,3], [4,4,4]]))

实现细节

该函数的作用是根据传入的数据positions和magnetization，将它们转换为Theano共享变量，以在模型运行过程中使用。如果verbose为True，函数将打印信息，说明共享变量的创建过程。共享变量的名称、类型和形状如下：

• 坐标共享变量：self.input_pos_shared
  • 名称：'input_pos_shared'
  • 类型：'float32'
  • 形状：(m, 3)
• 磁性共享变量：self.magnetization_tensor_shared
  • 名称：'magnetization_tensor_shared'
  • 类型：dtype
  • 形状：(m, 3)