本文整理匯總了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