本文整理汇总了Python中distarray.local.localarray.LocalArray.global_index[global_inds]方法的典型用法代码示例。如果您正苦于以下问题:Python LocalArray.global_index[global_inds]方法的具体用法?Python LocalArray.global_index[global_inds]怎么用?Python LocalArray.global_index[global_inds]使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类distarray.local.localarray.LocalArray
的用法示例。
在下文中一共展示了LocalArray.global_index[global_inds]方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_indexing_0
# 需要导入模块: from distarray.local.localarray import LocalArray [as 别名]
# 或者: from distarray.local.localarray.LocalArray import global_index[global_inds] [as 别名]
def test_indexing_0(self):
"""Can we get and set local elements for a simple dist?"""
distribution = Distribution.from_shape(comm=self.comm,
shape=(16, 16), dist=('b', 'n'))
a = LocalArray(distribution)
b = LocalArray(distribution)
for global_inds, value in ndenumerate(a):
a.global_index[global_inds] = 0.0
for global_inds, value in ndenumerate(a):
b.global_index[global_inds] = a.global_index[global_inds]
for i, value in ndenumerate(a):
self.assertEqual(b.global_index[i], a.global_index[i])
self.assertEqual(a.global_index[i], 0.0)
示例2: test_ndenumerate
# 需要导入模块: from distarray.local.localarray import LocalArray [as 别名]
# 或者: from distarray.local.localarray.LocalArray import global_index[global_inds] [as 别名]
def test_ndenumerate(self):
d = Distribution.from_shape(comm=self.comm,
shape=(16, 16, 2), dist=('c', 'b', 'n'))
a = LocalArray(d)
for global_inds, value in ndenumerate(a):
a.global_index[global_inds] = 0.0