mindspore.ops.sparse_segment_mean

mindspore.ops.sparse_segment_mean(x, indices, segment_ids)

и®Ўз®—иҫ“е…Ҙеј йҮҸзЁҖз–Ҹж®өзҡ„е№іеқҮеҖјгҖӮ

иӯҰе‘Ҡ

иҜҘжҺҘеҸЈд»Һ2.8.0зүҲжң¬ејҖе§Ӣе·Іиў«ејғз”ЁпјҢ并е°ҶеңЁ2.9.0пјҲдёҚеҗ«пјүд№ӢеҗҺзүҲжң¬дёӯ被移йҷӨгҖӮ

\[output_i = \frac{\sum_j x_{indices[j]}}{N}\]

е…¶дёӯ N дёәж»Ўи¶і \(segment\_ids[j] == i\) зҡ„е…ғзҙ дёӘж•°пјҢеҰӮжһң segment_ids дёӯжІЎжңүжҹҗдёӘеҖј i пјҢеҲҷ \(output[i] = 0\) гҖӮ

иҜҙжҳҺ

• еңЁCPUе№іеҸ°пјҢ segment_ids еҝ…йЎ»жҳҜеҚҮеәҸпјҢ indices еҝ…йЎ»еңЁ[0, x.shape[0])иҢғеӣҙеҶ…гҖӮ

• еңЁGPUе№іеҸ°пјҢеҜ№дәҺ segment_ids ж— йңҖжҺ’еәҸпјҢдҪҶеҸҜиғҪеҜјиҮҙжңӘе®ҡд№үзҡ„иЎҢдёәгҖӮи¶…еҮәиҢғеӣҙзҡ„ indices е°Ҷиў«еҝҪз•ҘгҖӮ

еҸӮж•°пјҡ

• x (Tensor) - иҫ“е…ҘtensorпјҢиҮіе°‘дёәдёҖз»ҙгҖӮ

• indices (Tensor) - жҢҮе®ҡзҙўеј•пјҢдёҖз»ҙtensorгҖӮ

• segment_ids (Tensor) - vpn永久免费梯子 дёҖз»ҙtensorпјҢеҝ…йЎ»жҢүеҚҮеәҸжҺ’еҲ—пјҢеҸҜеҢ…еҗ«йҮҚеӨҚеҖјгҖӮ

иҝ”еӣһпјҡ

Tensor

ж”ҜжҢҒе№іеҸ°пјҡ

Deprecated

ж ·дҫӢпјҡ

>>> import mindspore
>>> x = mindspore.tensor([[0, 1, 2], [1, 2, 3], [3, 6, 7]], vpn梯子 免费 dtype=mindspore.float32)
>>> indices = mindspore.tensor([0, 1, 2], dtype=mindspore.int32)
>>> segment_ids = 免费的vpn梯子 mindspore.tensor([1,2,2], dtype=mindspore.int32)
>>> out = mindspore.ops.sparse_segment_mean(x, indices, segment_ids)
>>> print(out)
[[0. 0. 0.]
 [0. 1. vpn梯子 免费 免费的vpn梯子 vpn free 2.]
 [2. 4. 5.]]