Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: b45a9f56c8
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b45a9f56c8'
${ noResults }
mindspore/tests/st/ops/gpu/test_gatherV2_op.py

1137 lines
48 kB
Raw Blame History 

  1. # Copyright 2019-2020 Huawei Technologies Co., Ltd

  2. #

  3. # Licensed under the Apache License, Version 2.0 (the "License");

  4. # you may not use this file except in compliance with the License.

  5. # You may obtain a copy of the License at

  6. #

  7. # http://www.apache.org/licenses/LICENSE-2.0

  8. #

  9. # Unless required by applicable law or agreed to in writing, software

  10. # distributed under the License is distributed on an "AS IS" BASIS,

  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  12. # See the License for the specific language governing permissions and

  13. # limitations under the License.

  14. # ============================================================================

  15. 

  16. import numpy as np

  17. import pytest

  18. 

  19. import mindspore.context as context

  20. import mindspore.nn as nn

  21. from mindspore import Tensor

  22. from mindspore.ops.operations import _inner_ops as inner

  23. from mindspore.ops import operations as P

  24. 

  25. 

  26. class GatherNet(nn.Cell):

  27.     def __init__(self):

  28.         super(GatherNet, self).__init__()

  29.         self.gather = P.Gather()

  30. 

  31.     def construct(self, x, indices):

  32.         return self.gather(x, indices, 1)

  33. 

  34. 

  35. @pytest.mark.level0

  36. @pytest.mark.platform_x86_gpu_training

  37. @pytest.mark.env_onecard

  38. def test_gather0():

  39.     x = Tensor(np.arange(2 * 3 * 4 * 5, dtype=np.float32).reshape(2, 3, 4, 5))

  40.     indices = Tensor(np.ones((2, 2, 4, 5), dtype='i4'))

  41.     expect = np.array([[[[[[[20., 21., 22., 23., 24.],

  42.                             [25., 26., 27., 28., 29.],

  43.                             [30., 31., 32., 33., 34.],

  44.                             [35., 36., 37., 38., 39.]],

  45. 

  46.                            [[20., 21., 22., 23., 24.],

  47.                             [25., 26., 27., 28., 29.],

  48.                             [30., 31., 32., 33., 34.],

  49.                             [35., 36., 37., 38., 39.]],

  50. 

  51.                            [[20., 21., 22., 23., 24.],

  52.                             [25., 26., 27., 28., 29.],

  53.                             [30., 31., 32., 33., 34.],

  54.                             [35., 36., 37., 38., 39.]],

  55. 

  56.                            [[20., 21., 22., 23., 24.],

  57.                             [25., 26., 27., 28., 29.],

  58.                             [30., 31., 32., 33., 34.],

  59.                             [35., 36., 37., 38., 39.]],

  60. 

  61.                            [[20., 21., 22., 23., 24.],

  62.                             [25., 26., 27., 28., 29.],

  63.                             [30., 31., 32., 33., 34.],

  64.                             [35., 36., 37., 38., 39.]]],

  65. 

  66.                           [[[20., 21., 22., 23., 24.],

  67.                             [25., 26., 27., 28., 29.],

  68.                             [30., 31., 32., 33., 34.],

  69.                             [35., 36., 37., 38., 39.]],

  70. 

  71.                            [[20., 21., 22., 23., 24.],

  72.                             [25., 26., 27., 28., 29.],

  73.                             [30., 31., 32., 33., 34.],

  74.                             [35., 36., 37., 38., 39.]],

  75. 

  76.                            [[20., 21., 22., 23., 24.],

  77.                             [25., 26., 27., 28., 29.],

  78.                             [30., 31., 32., 33., 34.],

  79.                             [35., 36., 37., 38., 39.]],

  80. 

  81.                            [[20., 21., 22., 23., 24.],

  82.                             [25., 26., 27., 28., 29.],

  83.                             [30., 31., 32., 33., 34.],

  84.                             [35., 36., 37., 38., 39.]],

  85. 

  86.                            [[20., 21., 22., 23., 24.],

  87.                             [25., 26., 27., 28., 29.],

  88.                             [30., 31., 32., 33., 34.],

  89.                             [35., 36., 37., 38., 39.]]],

  90. 

  91.                           [[[20., 21., 22., 23., 24.],

  92.                             [25., 26., 27., 28., 29.],

  93.                             [30., 31., 32., 33., 34.],

  94.                             [35., 36., 37., 38., 39.]],

  95. 

  96.                            [[20., 21., 22., 23., 24.],

  97.                             [25., 26., 27., 28., 29.],

  98.                             [30., 31., 32., 33., 34.],

  99.                             [35., 36., 37., 38., 39.]],

  100. 

  101.                            [[20., 21., 22., 23., 24.],

  102.                             [25., 26., 27., 28., 29.],

  103.                             [30., 31., 32., 33., 34.],

  104.                             [35., 36., 37., 38., 39.]],

  105. 

  106.                            [[20., 21., 22., 23., 24.],

  107.                             [25., 26., 27., 28., 29.],

  108.                             [30., 31., 32., 33., 34.],

  109.                             [35., 36., 37., 38., 39.]],

  110. 

  111.                            [[20., 21., 22., 23., 24.],

  112.                             [25., 26., 27., 28., 29.],

  113.                             [30., 31., 32., 33., 34.],

  114.                             [35., 36., 37., 38., 39.]]],

  115. 

  116.                           [[[20., 21., 22., 23., 24.],

  117.                             [25., 26., 27., 28., 29.],

  118.                             [30., 31., 32., 33., 34.],

  119.                             [35., 36., 37., 38., 39.]],

  120. 

  121.                            [[20., 21., 22., 23., 24.],

  122.                             [25., 26., 27., 28., 29.],

  123.                             [30., 31., 32., 33., 34.],

  124.                             [35., 36., 37., 38., 39.]],

  125. 

  126.                            [[20., 21., 22., 23., 24.],

  127.                             [25., 26., 27., 28., 29.],

  128.                             [30., 31., 32., 33., 34.],

  129.                             [35., 36., 37., 38., 39.]],

  130. 

  131.                            [[20., 21., 22., 23., 24.],

  132.                             [25., 26., 27., 28., 29.],

  133.                             [30., 31., 32., 33., 34.],

  134.                             [35., 36., 37., 38., 39.]],

  135. 

  136.                            [[20., 21., 22., 23., 24.],

  137.                             [25., 26., 27., 28., 29.],

  138.                             [30., 31., 32., 33., 34.],

  139.                             [35., 36., 37., 38., 39.]]]],

  140. 

  141.                          [[[[20., 21., 22., 23., 24.],

  142.                             [25., 26., 27., 28., 29.],

  143.                             [30., 31., 32., 33., 34.],

  144.                             [35., 36., 37., 38., 39.]],

  145. 

  146.                            [[20., 21., 22., 23., 24.],

  147.                             [25., 26., 27., 28., 29.],

  148.                             [30., 31., 32., 33., 34.],

  149.                             [35., 36., 37., 38., 39.]],

  150. 

  151.                            [[20., 21., 22., 23., 24.],

  152.                             [25., 26., 27., 28., 29.],

  153.                             [30., 31., 32., 33., 34.],

  154.                             [35., 36., 37., 38., 39.]],

  155. 

  156.                            [[20., 21., 22., 23., 24.],

  157.                             [25., 26., 27., 28., 29.],

  158.                             [30., 31., 32., 33., 34.],

  159.                             [35., 36., 37., 38., 39.]],

  160. 

  161.                            [[20., 21., 22., 23., 24.],

  162.                             [25., 26., 27., 28., 29.],

  163.                             [30., 31., 32., 33., 34.],

  164.                             [35., 36., 37., 38., 39.]]],

  165. 

  166.                           [[[20., 21., 22., 23., 24.],

  167.                             [25., 26., 27., 28., 29.],

  168.                             [30., 31., 32., 33., 34.],

  169.                             [35., 36., 37., 38., 39.]],

  170. 

  171.                            [[20., 21., 22., 23., 24.],

  172.                             [25., 26., 27., 28., 29.],

  173.                             [30., 31., 32., 33., 34.],

  174.                             [35., 36., 37., 38., 39.]],

  175. 

  176.                            [[20., 21., 22., 23., 24.],

  177.                             [25., 26., 27., 28., 29.],

  178.                             [30., 31., 32., 33., 34.],

  179.                             [35., 36., 37., 38., 39.]],

  180. 

  181.                            [[20., 21., 22., 23., 24.],

  182.                             [25., 26., 27., 28., 29.],

  183.                             [30., 31., 32., 33., 34.],

  184.                             [35., 36., 37., 38., 39.]],

  185. 

  186.                            [[20., 21., 22., 23., 24.],

  187.                             [25., 26., 27., 28., 29.],

  188.                             [30., 31., 32., 33., 34.],

  189.                             [35., 36., 37., 38., 39.]]],

  190. 

  191.                           [[[20., 21., 22., 23., 24.],

  192.                             [25., 26., 27., 28., 29.],

  193.                             [30., 31., 32., 33., 34.],

  194.                             [35., 36., 37., 38., 39.]],

  195. 

  196.                            [[20., 21., 22., 23., 24.],

  197.                             [25., 26., 27., 28., 29.],

  198.                             [30., 31., 32., 33., 34.],

  199.                             [35., 36., 37., 38., 39.]],

  200. 

  201.                            [[20., 21., 22., 23., 24.],

  202.                             [25., 26., 27., 28., 29.],

  203.                             [30., 31., 32., 33., 34.],

  204.                             [35., 36., 37., 38., 39.]],

  205. 

  206.                            [[20., 21., 22., 23., 24.],

  207.                             [25., 26., 27., 28., 29.],

  208.                             [30., 31., 32., 33., 34.],

  209.                             [35., 36., 37., 38., 39.]],

  210. 

  211.                            [[20., 21., 22., 23., 24.],

  212.                             [25., 26., 27., 28., 29.],

  213.                             [30., 31., 32., 33., 34.],

  214.                             [35., 36., 37., 38., 39.]]],

  215. 

  216.                           [[[20., 21., 22., 23., 24.],

  217.                             [25., 26., 27., 28., 29.],

  218.                             [30., 31., 32., 33., 34.],

  219.                             [35., 36., 37., 38., 39.]],

  220. 

  221.                            [[20., 21., 22., 23., 24.],

  222.                             [25., 26., 27., 28., 29.],

  223.                             [30., 31., 32., 33., 34.],

  224.                             [35., 36., 37., 38., 39.]],

  225. 

  226.                            [[20., 21., 22., 23., 24.],

  227.                             [25., 26., 27., 28., 29.],

  228.                             [30., 31., 32., 33., 34.],

  229.                             [35., 36., 37., 38., 39.]],

  230. 

  231.                            [[20., 21., 22., 23., 24.],

  232.                             [25., 26., 27., 28., 29.],

  233.                             [30., 31., 32., 33., 34.],

  234.                             [35., 36., 37., 38., 39.]],

  235. 

  236.                            [[20., 21., 22., 23., 24.],

  237.                             [25., 26., 27., 28., 29.],

  238.                             [30., 31., 32., 33., 34.],

  239.                             [35., 36., 37., 38., 39.]]]]],

  240. 

  241.                         [[[[[20., 21., 22., 23., 24.],

  242.                             [25., 26., 27., 28., 29.],

  243.                             [30., 31., 32., 33., 34.],

  244.                             [35., 36., 37., 38., 39.]],

  245. 

  246.                            [[20., 21., 22., 23., 24.],

  247.                             [25., 26., 27., 28., 29.],

  248.                             [30., 31., 32., 33., 34.],

  249.                             [35., 36., 37., 38., 39.]],

  250. 

  251.                            [[20., 21., 22., 23., 24.],

  252.                             [25., 26., 27., 28., 29.],

  253.                             [30., 31., 32., 33., 34.],

  254.                             [35., 36., 37., 38., 39.]],

  255. 

  256.                            [[20., 21., 22., 23., 24.],

  257.                             [25., 26., 27., 28., 29.],

  258.                             [30., 31., 32., 33., 34.],

  259.                             [35., 36., 37., 38., 39.]],

  260. 

  261.                            [[20., 21., 22., 23., 24.],

  262.                             [25., 26., 27., 28., 29.],

  263.                             [30., 31., 32., 33., 34.],

  264.                             [35., 36., 37., 38., 39.]]],

  265. 

  266.                           [[[20., 21., 22., 23., 24.],

  267.                             [25., 26., 27., 28., 29.],

  268.                             [30., 31., 32., 33., 34.],

  269.                             [35., 36., 37., 38., 39.]],

  270. 

  271.                            [[20., 21., 22., 23., 24.],

  272.                             [25., 26., 27., 28., 29.],

  273.                             [30., 31., 32., 33., 34.],

  274.                             [35., 36., 37., 38., 39.]],

  275. 

  276.                            [[20., 21., 22., 23., 24.],

  277.                             [25., 26., 27., 28., 29.],

  278.                             [30., 31., 32., 33., 34.],

  279.                             [35., 36., 37., 38., 39.]],

  280. 

  281.                            [[20., 21., 22., 23., 24.],

  282.                             [25., 26., 27., 28., 29.],

  283.                             [30., 31., 32., 33., 34.],

  284.                             [35., 36., 37., 38., 39.]],

  285. 

  286.                            [[20., 21., 22., 23., 24.],

  287.                             [25., 26., 27., 28., 29.],

  288.                             [30., 31., 32., 33., 34.],

  289.                             [35., 36., 37., 38., 39.]]],

  290. 

  291.                           [[[20., 21., 22., 23., 24.],

  292.                             [25., 26., 27., 28., 29.],

  293.                             [30., 31., 32., 33., 34.],

  294.                             [35., 36., 37., 38., 39.]],

  295. 

  296.                            [[20., 21., 22., 23., 24.],

  297.                             [25., 26., 27., 28., 29.],

  298.                             [30., 31., 32., 33., 34.],

  299.                             [35., 36., 37., 38., 39.]],

  300. 

  301.                            [[20., 21., 22., 23., 24.],

  302.                             [25., 26., 27., 28., 29.],

  303.                             [30., 31., 32., 33., 34.],

  304.                             [35., 36., 37., 38., 39.]],

  305. 

  306.                            [[20., 21., 22., 23., 24.],

  307.                             [25., 26., 27., 28., 29.],

  308.                             [30., 31., 32., 33., 34.],

  309.                             [35., 36., 37., 38., 39.]],

  310. 

  311.                            [[20., 21., 22., 23., 24.],

  312.                             [25., 26., 27., 28., 29.],

  313.                             [30., 31., 32., 33., 34.],

  314.                             [35., 36., 37., 38., 39.]]],

  315. 

  316.                           [[[20., 21., 22., 23., 24.],

  317.                             [25., 26., 27., 28., 29.],

  318.                             [30., 31., 32., 33., 34.],

  319.                             [35., 36., 37., 38., 39.]],

  320. 

  321.                            [[20., 21., 22., 23., 24.],

  322.                             [25., 26., 27., 28., 29.],

  323.                             [30., 31., 32., 33., 34.],

  324.                             [35., 36., 37., 38., 39.]],

  325. 

  326.                            [[20., 21., 22., 23., 24.],

  327.                             [25., 26., 27., 28., 29.],

  328.                             [30., 31., 32., 33., 34.],

  329.                             [35., 36., 37., 38., 39.]],

  330. 

  331.                            [[20., 21., 22., 23., 24.],

  332.                             [25., 26., 27., 28., 29.],

  333.                             [30., 31., 32., 33., 34.],

  334.                             [35., 36., 37., 38., 39.]],

  335. 

  336.                            [[20., 21., 22., 23., 24.],

  337.                             [25., 26., 27., 28., 29.],

  338.                             [30., 31., 32., 33., 34.],

  339.                             [35., 36., 37., 38., 39.]]]],

  340. 

  341.                          [[[[20., 21., 22., 23., 24.],

  342.                             [25., 26., 27., 28., 29.],

  343.                             [30., 31., 32., 33., 34.],

  344.                             [35., 36., 37., 38., 39.]],

  345. 

  346.                            [[20., 21., 22., 23., 24.],

  347.                             [25., 26., 27., 28., 29.],

  348.                             [30., 31., 32., 33., 34.],

  349.                             [35., 36., 37., 38., 39.]],

  350. 

  351.                            [[20., 21., 22., 23., 24.],

  352.                             [25., 26., 27., 28., 29.],

  353.                             [30., 31., 32., 33., 34.],

  354.                             [35., 36., 37., 38., 39.]],

  355. 

  356.                            [[20., 21., 22., 23., 24.],

  357.                             [25., 26., 27., 28., 29.],

  358.                             [30., 31., 32., 33., 34.],

  359.                             [35., 36., 37., 38., 39.]],

  360. 

  361.                            [[20., 21., 22., 23., 24.],

  362.                             [25., 26., 27., 28., 29.],

  363.                             [30., 31., 32., 33., 34.],

  364.                             [35., 36., 37., 38., 39.]]],

  365. 

  366.                           [[[20., 21., 22., 23., 24.],

  367.                             [25., 26., 27., 28., 29.],

  368.                             [30., 31., 32., 33., 34.],

  369.                             [35., 36., 37., 38., 39.]],

  370. 

  371.                            [[20., 21., 22., 23., 24.],

  372.                             [25., 26., 27., 28., 29.],

  373.                             [30., 31., 32., 33., 34.],

  374.                             [35., 36., 37., 38., 39.]],

  375. 

  376.                            [[20., 21., 22., 23., 24.],

  377.                             [25., 26., 27., 28., 29.],

  378.                             [30., 31., 32., 33., 34.],

  379.                             [35., 36., 37., 38., 39.]],

  380. 

  381.                            [[20., 21., 22., 23., 24.],

  382.                             [25., 26., 27., 28., 29.],

  383.                             [30., 31., 32., 33., 34.],

  384.                             [35., 36., 37., 38., 39.]],

  385. 

  386.                            [[20., 21., 22., 23., 24.],

  387.                             [25., 26., 27., 28., 29.],

  388.                             [30., 31., 32., 33., 34.],

  389.                             [35., 36., 37., 38., 39.]]],

  390. 

  391.                           [[[20., 21., 22., 23., 24.],

  392.                             [25., 26., 27., 28., 29.],

  393.                             [30., 31., 32., 33., 34.],

  394.                             [35., 36., 37., 38., 39.]],

  395. 

  396.                            [[20., 21., 22., 23., 24.],

  397.                             [25., 26., 27., 28., 29.],

  398.                             [30., 31., 32., 33., 34.],

  399.                             [35., 36., 37., 38., 39.]],

  400. 

  401.                            [[20., 21., 22., 23., 24.],

  402.                             [25., 26., 27., 28., 29.],

  403.                             [30., 31., 32., 33., 34.],

  404.                             [35., 36., 37., 38., 39.]],

  405. 

  406.                            [[20., 21., 22., 23., 24.],

  407.                             [25., 26., 27., 28., 29.],

  408.                             [30., 31., 32., 33., 34.],

  409.                             [35., 36., 37., 38., 39.]],

  410. 

  411.                            [[20., 21., 22., 23., 24.],

  412.                             [25., 26., 27., 28., 29.],

  413.                             [30., 31., 32., 33., 34.],

  414.                             [35., 36., 37., 38., 39.]]],

  415. 

  416.                           [[[20., 21., 22., 23., 24.],

  417.                             [25., 26., 27., 28., 29.],

  418.                             [30., 31., 32., 33., 34.],

  419.                             [35., 36., 37., 38., 39.]],

  420. 

  421.                            [[20., 21., 22., 23., 24.],

  422.                             [25., 26., 27., 28., 29.],

  423.                             [30., 31., 32., 33., 34.],

  424.                             [35., 36., 37., 38., 39.]],

  425. 

  426.                            [[20., 21., 22., 23., 24.],

  427.                             [25., 26., 27., 28., 29.],

  428.                             [30., 31., 32., 33., 34.],

  429.                             [35., 36., 37., 38., 39.]],

  430. 

  431.                            [[20., 21., 22., 23., 24.],

  432.                             [25., 26., 27., 28., 29.],

  433.                             [30., 31., 32., 33., 34.],

  434.                             [35., 36., 37., 38., 39.]],

  435. 

  436.                            [[20., 21., 22., 23., 24.],

  437.                             [25., 26., 27., 28., 29.],

  438.                             [30., 31., 32., 33., 34.],

  439.                             [35., 36., 37., 38., 39.]]]]]],

  440. 

  441.                        [[[[[[80., 81., 82., 83., 84.],

  442.                             [85., 86., 87., 88., 89.],

  443.                             [90., 91., 92., 93., 94.],

  444.                             [95., 96., 97., 98., 99.]],

  445. 

  446.                            [[80., 81., 82., 83., 84.],

  447.                             [85., 86., 87., 88., 89.],

  448.                             [90., 91., 92., 93., 94.],

  449.                             [95., 96., 97., 98., 99.]],

  450. 

  451.                            [[80., 81., 82., 83., 84.],

  452.                             [85., 86., 87., 88., 89.],

  453.                             [90., 91., 92., 93., 94.],

  454.                             [95., 96., 97., 98., 99.]],

  455. 

  456.                            [[80., 81., 82., 83., 84.],

  457.                             [85., 86., 87., 88., 89.],

  458.                             [90., 91., 92., 93., 94.],

  459.                             [95., 96., 97., 98., 99.]],

  460. 

  461.                            [[80., 81., 82., 83., 84.],

  462.                             [85., 86., 87., 88., 89.],

  463.                             [90., 91., 92., 93., 94.],

  464.                             [95., 96., 97., 98., 99.]]],

  465. 

  466.                           [[[80., 81., 82., 83., 84.],

  467.                             [85., 86., 87., 88., 89.],

  468.                             [90., 91., 92., 93., 94.],

  469.                             [95., 96., 97., 98., 99.]],

  470. 

  471.                            [[80., 81., 82., 83., 84.],

  472.                             [85., 86., 87., 88., 89.],

  473.                             [90., 91., 92., 93., 94.],

  474.                             [95., 96., 97., 98., 99.]],

  475. 

  476.                            [[80., 81., 82., 83., 84.],

  477.                             [85., 86., 87., 88., 89.],

  478.                             [90., 91., 92., 93., 94.],

  479.                             [95., 96., 97., 98., 99.]],

  480. 

  481.                            [[80., 81., 82., 83., 84.],

  482.                             [85., 86., 87., 88., 89.],

  483.                             [90., 91., 92., 93., 94.],

  484.                             [95., 96., 97., 98., 99.]],

  485. 

  486.                            [[80., 81., 82., 83., 84.],

  487.                             [85., 86., 87., 88., 89.],

  488.                             [90., 91., 92., 93., 94.],

  489.                             [95., 96., 97., 98., 99.]]],

  490. 

  491.                           [[[80., 81., 82., 83., 84.],

  492.                             [85., 86., 87., 88., 89.],

  493.                             [90., 91., 92., 93., 94.],

  494.                             [95., 96., 97., 98., 99.]],

  495. 

  496.                            [[80., 81., 82., 83., 84.],

  497.                             [85., 86., 87., 88., 89.],

  498.                             [90., 91., 92., 93., 94.],

  499.                             [95., 96., 97., 98., 99.]],

  500. 

  501.                            [[80., 81., 82., 83., 84.],

  502.                             [85., 86., 87., 88., 89.],

  503.                             [90., 91., 92., 93., 94.],

  504.                             [95., 96., 97., 98., 99.]],

  505. 

  506.                            [[80., 81., 82., 83., 84.],

  507.                             [85., 86., 87., 88., 89.],

  508.                             [90., 91., 92., 93., 94.],

  509.                             [95., 96., 97., 98., 99.]],

  510. 

  511.                            [[80., 81., 82., 83., 84.],

  512.                             [85., 86., 87., 88., 89.],

  513.                             [90., 91., 92., 93., 94.],

  514.                             [95., 96., 97., 98., 99.]]],

  515. 

  516.                           [[[80., 81., 82., 83., 84.],

  517.                             [85., 86., 87., 88., 89.],

  518.                             [90., 91., 92., 93., 94.],

  519.                             [95., 96., 97., 98., 99.]],

  520. 

  521.                            [[80., 81., 82., 83., 84.],

  522.                             [85., 86., 87., 88., 89.],

  523.                             [90., 91., 92., 93., 94.],

  524.                             [95., 96., 97., 98., 99.]],

  525. 

  526.                            [[80., 81., 82., 83., 84.],

  527.                             [85., 86., 87., 88., 89.],

  528.                             [90., 91., 92., 93., 94.],

  529.                             [95., 96., 97., 98., 99.]],

  530. 

  531.                            [[80., 81., 82., 83., 84.],

  532.                             [85., 86., 87., 88., 89.],

  533.                             [90., 91., 92., 93., 94.],

  534.                             [95., 96., 97., 98., 99.]],

  535. 

  536.                            [[80., 81., 82., 83., 84.],

  537.                             [85., 86., 87., 88., 89.],

  538.                             [90., 91., 92., 93., 94.],

  539.                             [95., 96., 97., 98., 99.]]]],

  540. 

  541.                          [[[[80., 81., 82., 83., 84.],

  542.                             [85., 86., 87., 88., 89.],

  543.                             [90., 91., 92., 93., 94.],

  544.                             [95., 96., 97., 98., 99.]],

  545. 

  546.                            [[80., 81., 82., 83., 84.],

  547.                             [85., 86., 87., 88., 89.],

  548.                             [90., 91., 92., 93., 94.],

  549.                             [95., 96., 97., 98., 99.]],

  550. 

  551.                            [[80., 81., 82., 83., 84.],

  552.                             [85., 86., 87., 88., 89.],

  553.                             [90., 91., 92., 93., 94.],

  554.                             [95., 96., 97., 98., 99.]],

  555. 

  556.                            [[80., 81., 82., 83., 84.],

  557.                             [85., 86., 87., 88., 89.],

  558.                             [90., 91., 92., 93., 94.],

  559.                             [95., 96., 97., 98., 99.]],

  560. 

  561.                            [[80., 81., 82., 83., 84.],

  562.                             [85., 86., 87., 88., 89.],

  563.                             [90., 91., 92., 93., 94.],

  564.                             [95., 96., 97., 98., 99.]]],

  565. 

  566.                           [[[80., 81., 82., 83., 84.],

  567.                             [85., 86., 87., 88., 89.],

  568.                             [90., 91., 92., 93., 94.],

  569.                             [95., 96., 97., 98., 99.]],

  570. 

  571.                            [[80., 81., 82., 83., 84.],

  572.                             [85., 86., 87., 88., 89.],

  573.                             [90., 91., 92., 93., 94.],

  574.                             [95., 96., 97., 98., 99.]],

  575. 

  576.                            [[80., 81., 82., 83., 84.],

  577.                             [85., 86., 87., 88., 89.],

  578.                             [90., 91., 92., 93., 94.],

  579.                             [95., 96., 97., 98., 99.]],

  580. 

  581.                            [[80., 81., 82., 83., 84.],

  582.                             [85., 86., 87., 88., 89.],

  583.                             [90., 91., 92., 93., 94.],

  584.                             [95., 96., 97., 98., 99.]],

  585. 

  586.                            [[80., 81., 82., 83., 84.],

  587.                             [85., 86., 87., 88., 89.],

  588.                             [90., 91., 92., 93., 94.],

  589.                             [95., 96., 97., 98., 99.]]],

  590. 

  591.                           [[[80., 81., 82., 83., 84.],

  592.                             [85., 86., 87., 88., 89.],

  593.                             [90., 91., 92., 93., 94.],

  594.                             [95., 96., 97., 98., 99.]],

  595. 

  596.                            [[80., 81., 82., 83., 84.],

  597.                             [85., 86., 87., 88., 89.],

  598.                             [90., 91., 92., 93., 94.],

  599.                             [95., 96., 97., 98., 99.]],

  600. 

  601.                            [[80., 81., 82., 83., 84.],

  602.                             [85., 86., 87., 88., 89.],

  603.                             [90., 91., 92., 93., 94.],

  604.                             [95., 96., 97., 98., 99.]],

  605. 

  606.                            [[80., 81., 82., 83., 84.],

  607.                             [85., 86., 87., 88., 89.],

  608.                             [90., 91., 92., 93., 94.],

  609.                             [95., 96., 97., 98., 99.]],

  610. 

  611.                            [[80., 81., 82., 83., 84.],

  612.                             [85., 86., 87., 88., 89.],

  613.                             [90., 91., 92., 93., 94.],

  614.                             [95., 96., 97., 98., 99.]]],

  615. 

  616.                           [[[80., 81., 82., 83., 84.],

  617.                             [85., 86., 87., 88., 89.],

  618.                             [90., 91., 92., 93., 94.],

  619.                             [95., 96., 97., 98., 99.]],

  620. 

  621.                            [[80., 81., 82., 83., 84.],

  622.                             [85., 86., 87., 88., 89.],

  623.                             [90., 91., 92., 93., 94.],

  624.                             [95., 96., 97., 98., 99.]],

  625. 

  626.                            [[80., 81., 82., 83., 84.],

  627.                             [85., 86., 87., 88., 89.],

  628.                             [90., 91., 92., 93., 94.],

  629.                             [95., 96., 97., 98., 99.]],

  630. 

  631.                            [[80., 81., 82., 83., 84.],

  632.                             [85., 86., 87., 88., 89.],

  633.                             [90., 91., 92., 93., 94.],

  634.                             [95., 96., 97., 98., 99.]],

  635. 

  636.                            [[80., 81., 82., 83., 84.],

  637.                             [85., 86., 87., 88., 89.],

  638.                             [90., 91., 92., 93., 94.],

  639.                             [95., 96., 97., 98., 99.]]]]],

  640. 

  641.                         [[[[[80., 81., 82., 83., 84.],

  642.                             [85., 86., 87., 88., 89.],

  643.                             [90., 91., 92., 93., 94.],

  644.                             [95., 96., 97., 98., 99.]],

  645. 

  646.                            [[80., 81., 82., 83., 84.],

  647.                             [85., 86., 87., 88., 89.],

  648.                             [90., 91., 92., 93., 94.],

  649.                             [95., 96., 97., 98., 99.]],

  650. 

  651.                            [[80., 81., 82., 83., 84.],

  652.                             [85., 86., 87., 88., 89.],

  653.                             [90., 91., 92., 93., 94.],

  654.                             [95., 96., 97., 98., 99.]],

  655. 

  656.                            [[80., 81., 82., 83., 84.],

  657.                             [85., 86., 87., 88., 89.],

  658.                             [90., 91., 92., 93., 94.],

  659.                             [95., 96., 97., 98., 99.]],

  660. 

  661.                            [[80., 81., 82., 83., 84.],

  662.                             [85., 86., 87., 88., 89.],

  663.                             [90., 91., 92., 93., 94.],

  664.                             [95., 96., 97., 98., 99.]]],

  665. 

  666.                           [[[80., 81., 82., 83., 84.],

  667.                             [85., 86., 87., 88., 89.],

  668.                             [90., 91., 92., 93., 94.],

  669.                             [95., 96., 97., 98., 99.]],

  670. 

  671.                            [[80., 81., 82., 83., 84.],

  672.                             [85., 86., 87., 88., 89.],

  673.                             [90., 91., 92., 93., 94.],

  674.                             [95., 96., 97., 98., 99.]],

  675. 

  676.                            [[80., 81., 82., 83., 84.],

  677.                             [85., 86., 87., 88., 89.],

  678.                             [90., 91., 92., 93., 94.],

  679.                             [95., 96., 97., 98., 99.]],

  680. 

  681.                            [[80., 81., 82., 83., 84.],

  682.                             [85., 86., 87., 88., 89.],

  683.                             [90., 91., 92., 93., 94.],

  684.                             [95., 96., 97., 98., 99.]],

  685. 

  686.                            [[80., 81., 82., 83., 84.],

  687.                             [85., 86., 87., 88., 89.],

  688.                             [90., 91., 92., 93., 94.],

  689.                             [95., 96., 97., 98., 99.]]],

  690. 

  691.                           [[[80., 81., 82., 83., 84.],

  692.                             [85., 86., 87., 88., 89.],

  693.                             [90., 91., 92., 93., 94.],

  694.                             [95., 96., 97., 98., 99.]],

  695. 

  696.                            [[80., 81., 82., 83., 84.],

  697.                             [85., 86., 87., 88., 89.],

  698.                             [90., 91., 92., 93., 94.],

  699.                             [95., 96., 97., 98., 99.]],

  700. 

  701.                            [[80., 81., 82., 83., 84.],

  702.                             [85., 86., 87., 88., 89.],

  703.                             [90., 91., 92., 93., 94.],

  704.                             [95., 96., 97., 98., 99.]],

  705. 

  706.                            [[80., 81., 82., 83., 84.],

  707.                             [85., 86., 87., 88., 89.],

  708.                             [90., 91., 92., 93., 94.],

  709.                             [95., 96., 97., 98., 99.]],

  710. 

  711.                            [[80., 81., 82., 83., 84.],

  712.                             [85., 86., 87., 88., 89.],

  713.                             [90., 91., 92., 93., 94.],

  714.                             [95., 96., 97., 98., 99.]]],

  715. 

  716.                           [[[80., 81., 82., 83., 84.],

  717.                             [85., 86., 87., 88., 89.],

  718.                             [90., 91., 92., 93., 94.],

  719.                             [95., 96., 97., 98., 99.]],

  720. 

  721.                            [[80., 81., 82., 83., 84.],

  722.                             [85., 86., 87., 88., 89.],

  723.                             [90., 91., 92., 93., 94.],

  724.                             [95., 96., 97., 98., 99.]],

  725. 

  726.                            [[80., 81., 82., 83., 84.],

  727.                             [85., 86., 87., 88., 89.],

  728.                             [90., 91., 92., 93., 94.],

  729.                             [95., 96., 97., 98., 99.]],

  730. 

  731.                            [[80., 81., 82., 83., 84.],

  732.                             [85., 86., 87., 88., 89.],

  733.                             [90., 91., 92., 93., 94.],

  734.                             [95., 96., 97., 98., 99.]],

  735. 

  736.                            [[80., 81., 82., 83., 84.],

  737.                             [85., 86., 87., 88., 89.],

  738.                             [90., 91., 92., 93., 94.],

  739.                             [95., 96., 97., 98., 99.]]]],

  740. 

  741.                          [[[[80., 81., 82., 83., 84.],

  742.                             [85., 86., 87., 88., 89.],

  743.                             [90., 91., 92., 93., 94.],

  744.                             [95., 96., 97., 98., 99.]],

  745. 

  746.                            [[80., 81., 82., 83., 84.],

  747.                             [85., 86., 87., 88., 89.],

  748.                             [90., 91., 92., 93., 94.],

  749.                             [95., 96., 97., 98., 99.]],

  750. 

  751.                            [[80., 81., 82., 83., 84.],

  752.                             [85., 86., 87., 88., 89.],

  753.                             [90., 91., 92., 93., 94.],

  754.                             [95., 96., 97., 98., 99.]],

  755. 

  756.                            [[80., 81., 82., 83., 84.],

  757.                             [85., 86., 87., 88., 89.],

  758.                             [90., 91., 92., 93., 94.],

  759.                             [95., 96., 97., 98., 99.]],

  760. 

  761.                            [[80., 81., 82., 83., 84.],

  762.                             [85., 86., 87., 88., 89.],

  763.                             [90., 91., 92., 93., 94.],

  764.                             [95., 96., 97., 98., 99.]]],

  765. 

  766.                           [[[80., 81., 82., 83., 84.],

  767.                             [85., 86., 87., 88., 89.],

  768.                             [90., 91., 92., 93., 94.],

  769.                             [95., 96., 97., 98., 99.]],

  770. 

  771.                            [[80., 81., 82., 83., 84.],

  772.                             [85., 86., 87., 88., 89.],

  773.                             [90., 91., 92., 93., 94.],

  774.                             [95., 96., 97., 98., 99.]],

  775. 

  776.                            [[80., 81., 82., 83., 84.],

  777.                             [85., 86., 87., 88., 89.],

  778.                             [90., 91., 92., 93., 94.],

  779.                             [95., 96., 97., 98., 99.]],

  780. 

  781.                            [[80., 81., 82., 83., 84.],

  782.                             [85., 86., 87., 88., 89.],

  783.                             [90., 91., 92., 93., 94.],

  784.                             [95., 96., 97., 98., 99.]],

  785. 

  786.                            [[80., 81., 82., 83., 84.],

  787.                             [85., 86., 87., 88., 89.],

  788.                             [90., 91., 92., 93., 94.],

  789.                             [95., 96., 97., 98., 99.]]],

  790. 

  791.                           [[[80., 81., 82., 83., 84.],

  792.                             [85., 86., 87., 88., 89.],

  793.                             [90., 91., 92., 93., 94.],

  794.                             [95., 96., 97., 98., 99.]],

  795. 

  796.                            [[80., 81., 82., 83., 84.],

  797.                             [85., 86., 87., 88., 89.],

  798.                             [90., 91., 92., 93., 94.],

  799.                             [95., 96., 97., 98., 99.]],

  800. 

  801.                            [[80., 81., 82., 83., 84.],

  802.                             [85., 86., 87., 88., 89.],

  803.                             [90., 91., 92., 93., 94.],

  804.                             [95., 96., 97., 98., 99.]],

  805. 

  806.                            [[80., 81., 82., 83., 84.],

  807.                             [85., 86., 87., 88., 89.],

  808.                             [90., 91., 92., 93., 94.],

  809.                             [95., 96., 97., 98., 99.]],

  810. 

  811.                            [[80., 81., 82., 83., 84.],

  812.                             [85., 86., 87., 88., 89.],

  813.                             [90., 91., 92., 93., 94.],

  814.                             [95., 96., 97., 98., 99.]]],

  815. 

  816.                           [[[80., 81., 82., 83., 84.],

  817.                             [85., 86., 87., 88., 89.],

  818.                             [90., 91., 92., 93., 94.],

  819.                             [95., 96., 97., 98., 99.]],

  820. 

  821.                            [[80., 81., 82., 83., 84.],

  822.                             [85., 86., 87., 88., 89.],

  823.                             [90., 91., 92., 93., 94.],

  824.                             [95., 96., 97., 98., 99.]],

  825. 

  826.                            [[80., 81., 82., 83., 84.],

  827.                             [85., 86., 87., 88., 89.],

  828.                             [90., 91., 92., 93., 94.],

  829.                             [95., 96., 97., 98., 99.]],

  830. 

  831.                            [[80., 81., 82., 83., 84.],

  832.                             [85., 86., 87., 88., 89.],

  833.                             [90., 91., 92., 93., 94.],

  834.                             [95., 96., 97., 98., 99.]],

  835. 

  836.                            [[80., 81., 82., 83., 84.],

  837.                             [85., 86., 87., 88., 89.],

  838.                             [90., 91., 92., 93., 94.],

  839.                             [95., 96., 97., 98., 99.]]]]]]])

  840. 

  841.     context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

  842.     gather = GatherNet()

  843.     output = gather(x, indices)

  844.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  845.     diff = output.asnumpy() - expect

  846.     assert np.all(diff < error)

  847.     assert np.all(-diff < error)

  848. 

  849. 

  850. class GatherNet1(nn.Cell):

  851.     def __init__(self):

  852.         super(GatherNet1, self).__init__()

  853.         self.gather = P.Gather()

  854. 

  855.     def construct(self, x, indices):

  856.         return self.gather(x, indices, -1)

  857. 

  858. 

  859. @pytest.mark.level0

  860. @pytest.mark.platform_x86_gpu_training

  861. @pytest.mark.env_onecard

  862. def test_gather1():

  863.     x = Tensor(np.arange(2 * 3 * 4 * 5, dtype=np.float32).reshape(2, 3, 4, 5))

  864.     indices = Tensor(np.array([1, 3, 4], dtype='i4'))

  865.     expect = np.array([[[[1., 3., 4.],

  866.                          [6., 8., 9.],

  867.                          [11., 13., 14.],

  868.                          [16., 18., 19.]],

  869. 

  870.                         [[21., 23., 24.],

  871.                          [26., 28., 29.],

  872.                          [31., 33., 34.],

  873.                          [36., 38., 39.]],

  874. 

  875.                         [[41., 43., 44.],

  876.                          [46., 48., 49.],

  877.                          [51., 53., 54.],

  878.                          [56., 58., 59.]]],

  879. 

  880.                        [[[61., 63., 64.],

  881.                          [66., 68., 69.],

  882.                          [71., 73., 74.],

  883.                          [76., 78., 79.]],

  884. 

  885.                         [[81., 83., 84.],

  886.                          [86., 88., 89.],

  887.                          [91., 93., 94.],

  888.                          [96., 98., 99.]],

  889. 

  890.                         [[101., 103., 104.],

  891.                          [106., 108., 109.],

  892.                          [111., 113., 114.],

  893.                          [116., 118., 119.]]]])

  894. 

  895.     context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

  896.     gather = GatherNet1()

  897.     output = gather(x, indices)

  898.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  899.     diff = output.asnumpy() - expect

  900.     assert np.all(diff < error)

  901.     assert np.all(-diff < error)

  902. 

  903. 

  904. class GatherNet2(nn.Cell):

  905.     def __init__(self):

  906.         super(GatherNet2, self).__init__()

  907.         self.gather = P.Gather()

  908. 

  909.     def construct(self, x, indices):

  910.         return self.gather(x, indices, 0)

  911. 

  912. 

  913. @pytest.mark.level0

  914. @pytest.mark.platform_x86_gpu_training

  915. @pytest.mark.env_onecard

  916. def test_gather2():

  917.     x = Tensor(np.array([[4., 5., 4., 1., 5.,],

  918.                          [4., 9., 5., 6., 4.,],

  919.                          [9., 8., 4., 3., 6.,],

  920.                          [0., 4., 2., 2., 8.,],

  921.                          [1., 8., 6., 2., 8.,],

  922.                          [8., 1., 9., 7., 3.,],

  923.                          [7., 9., 2., 5., 7.,],

  924.                          [9., 8., 6., 8., 5.,],

  925.                          [3., 7., 2., 7., 4.,],

  926.                          [4., 2., 8., 2., 9.,]]

  927.                         ).astype(np.float32))

  928. 

  929.     indices = Tensor(np.array([[4000, 1, 300000]]).astype(np.int64))

  930.     expect = np.array([[[0., 0., 0., 0., 0.],

  931.                         [4., 9., 5., 6., 4.],

  932.                         [0., 0., 0., 0., 0.]]])

  933. 

  934.     context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

  935.     gather = GatherNet2()

  936.     output = gather(x, indices)

  937.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  938.     diff = output.asnumpy() - expect

  939.     assert np.all(diff < error)

  940.     assert np.all(-diff < error)

  941. 

  942. 

  943. # Dynamic Shape testing ahead

  944. class GatherNetDynamic(nn.Cell):

  945.     def __init__(self, axis=0, dyn_a=True, dyn_b=True):

  946.         super(GatherNetDynamic, self).__init__()

  947.         self.gather = P.Gather()

  948.         self.gpu_convert_to_dynamic_shape = inner.GpuConvertToDynamicShape()

  949.         self.to_dyn_1 = dyn_a

  950.         self.to_dyn_2 = dyn_b

  951.         self.axis = axis

  952.     def construct(self, x, indices):

  953.         # testing selective inputs being dynamic

  954.         if self.to_dyn_1:

  955.             x = self.gpu_convert_to_dynamic_shape(x)

  956.         if self.to_dyn_2:

  957.             indices = self.gpu_convert_to_dynamic_shape(indices)

  958.         return self.gather(x, indices, self.axis)

  959. 

  960. 

  961. @pytest.mark.level0

  962. @pytest.mark.platform_x86_gpu_training

  963. @pytest.mark.env_onecard

  964. def test_gatherV2_dyn_ab():

  965.     """

  966.     Tests for Dynamic shape with both inputs dynamic

  967.     """

  968.     context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

  969.     gather = GatherNetDynamic()

  970.     x = Tensor(np.array([[4., 5., 4., 1., 5.,],

  971.                          [4., 9., 5., 6., 4.,],

  972.                          [9., 8., 4., 3., 6.,],

  973.                          [0., 4., 2., 2., 8.,],

  974.                          [1., 8., 6., 2., 8.,],

  975.                          [8., 1., 9., 7., 3.,],

  976.                          [7., 9., 2., 5., 7.,],

  977.                          [9., 8., 6., 8., 5.,],

  978.                          [3., 7., 2., 7., 4.,],

  979.                          [4., 2., 8., 2., 9.,]]

  980.                         ).astype(np.float32))

  981.     indices = Tensor(np.array([[4000, 1, 300000]]).astype(np.int32))

  982.     expect = np.array([[[0., 0., 0., 0., 0.],

  983.                         [4., 9., 5., 6., 4.],

  984.                         [0., 0., 0., 0., 0.]]])

  985.     output = gather(x, indices)

  986.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  987.     diff = output.asnumpy() - expect

  988.     assert np.all(diff < error)

  989.     assert np.all(-diff < error)

  990. 

  991. 

  992. @pytest.mark.level0

  993. @pytest.mark.platform_x86_gpu_training

  994. @pytest.mark.env_onecard

  995. def test_gatherV2_dyn_a():

  996.     """

  997.     Tests for Dynamic shape with only first input dynamic

  998.     """

  999.     context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

  1000.     gather = GatherNetDynamic(-1, True, False)

  1001.     # test 1

  1002.     x = Tensor(np.array([[4., 5., 4., 1., 5.,],

  1003.                          [4., 9., 5., 6., 4.,],

  1004.                          [9., 8., 4., 3., 6.,],

  1005.                          [0., 4., 2., 2., 8.,],

  1006.                          [1., 8., 6., 2., 8.,],

  1007.                          [8., 1., 9., 7., 3.,],

  1008.                          [7., 9., 2., 5., 7.,],

  1009.                          [9., 8., 6., 8., 5.,],

  1010.                          [3., 7., 2., 7., 4.,],

  1011.                          [4., 2., 8., 2., 9.,]]

  1012.                         ).astype(np.float32))

  1013.     indices = Tensor(np.array([[4000, 1, 300000]]).astype(np.int64))

  1014.     expect = np.array([[[0., 5., 0.]],

  1015.                        [[0., 9., 0.]],

  1016.                        [[0., 8., 0.]],

  1017.                        [[0., 4., 0.]],

  1018.                        [[0., 8., 0.]],

  1019.                        [[0., 1., 0.]],

  1020.                        [[0., 9., 0.]],

  1021.                        [[0., 8., 0.]],

  1022.                        [[0., 7., 0.]],

  1023.                        [[0., 2., 0.]]]).astype(np.float32)

  1024.     output = gather(x, indices)

  1025.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  1026.     diff = output.asnumpy() - expect

  1027.     assert np.all(diff < error)

  1028.     assert np.all(-diff < error)

  1029.     # test 2

  1030.     x = Tensor(np.arange(2 * 3 * 4 * 5, dtype=np.float32).reshape(2, 3, 4, 5))

  1031.     indices = Tensor(np.array([1, 3, 4], dtype='i4'))

  1032.     expect = np.array([[[[1., 3., 4.],

  1033.                          [6., 8., 9.],

  1034.                          [11., 13., 14.],

  1035.                          [16., 18., 19.]],

  1036. 

  1037.                         [[21., 23., 24.],

  1038.                          [26., 28., 29.],

  1039.                          [31., 33., 34.],

  1040.                          [36., 38., 39.]],

  1041. 

  1042.                         [[41., 43., 44.],

  1043.                          [46., 48., 49.],

  1044.                          [51., 53., 54.],

  1045.                          [56., 58., 59.]]],

  1046. 

  1047.                        [[[61., 63., 64.],

  1048.                          [66., 68., 69.],

  1049.                          [71., 73., 74.],

  1050.                          [76., 78., 79.]],

  1051. 

  1052.                         [[81., 83., 84.],

  1053.                          [86., 88., 89.],

  1054.                          [91., 93., 94.],

  1055.                          [96., 98., 99.]],

  1056. 

  1057.                         [[101., 103., 104.],

  1058.                          [106., 108., 109.],

  1059.                          [111., 113., 114.],

  1060.                          [116., 118., 119.]]]])

  1061.     output = gather(x, indices)

  1062.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  1063.     diff = output.asnumpy() - expect

  1064.     assert np.all(diff < error)

  1065.     assert np.all(-diff < error)

  1066. 

  1067. 

  1068. @pytest.mark.level0

  1069. @pytest.mark.platform_x86_gpu_training

  1070. @pytest.mark.env_onecard

  1071. def test_gatherV2_dyn_b():

  1072.     """

  1073.     Tests for Dynamic shape with only second input dynamic

  1074.     """

  1075.     context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

  1076.     gather = GatherNetDynamic(-1, False, True)

  1077.     # test 1

  1078.     x = Tensor(np.array([[4., 5., 4., 1., 5.,],

  1079.                          [4., 9., 5., 6., 4.,],

  1080.                          [9., 8., 4., 3., 6.,],

  1081.                          [0., 4., 2., 2., 8.,],

  1082.                          [1., 8., 6., 2., 8.,],

  1083.                          [8., 1., 9., 7., 3.,],

  1084.                          [7., 9., 2., 5., 7.,],

  1085.                          [9., 8., 6., 8., 5.,],

  1086.                          [3., 7., 2., 7., 4.,],

  1087.                          [4., 2., 8., 2., 9.,]]

  1088.                         ).astype(np.float32))

  1089.     indices = Tensor(np.array([[4000, 1, 300000]]).astype(np.int32))

  1090.     expect = np.array([[[0., 5., 0.]],

  1091.                        [[0., 9., 0.]],

  1092.                        [[0., 8., 0.]],

  1093.                        [[0., 4., 0.]],

  1094.                        [[0., 8., 0.]],

  1095.                        [[0., 1., 0.]],

  1096.                        [[0., 9., 0.]],

  1097.                        [[0., 8., 0.]],

  1098.                        [[0., 7., 0.]],

  1099.                        [[0., 2., 0.]]]).astype(np.float32)

  1100.     output = gather(x, indices)

  1101.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  1102.     diff = output.asnumpy() - expect

  1103.     assert np.all(diff < error)

  1104.     assert np.all(-diff < error)

  1105.     # test 2

  1106.     x = Tensor(np.arange(2 * 3 * 4 * 5, dtype=np.float32).reshape(2, 3, 4, 5))

  1107.     indices = Tensor(np.array([1, 3, 4], dtype='i4'))

  1108.     expect = np.array([[[[1., 3., 4.],

  1109.                          [6., 8., 9.],

  1110.                          [11., 13., 14.],

  1111.                          [16., 18., 19.]],

  1112.                         [[21., 23., 24.],

  1113.                          [26., 28., 29.],

  1114.                          [31., 33., 34.],

  1115.                          [36., 38., 39.]],

  1116.                         [[41., 43., 44.],

  1117.                          [46., 48., 49.],

  1118.                          [51., 53., 54.],

  1119.                          [56., 58., 59.]]],

  1120.                        [[[61., 63., 64.],

  1121.                          [66., 68., 69.],

  1122.                          [71., 73., 74.],

  1123.                          [76., 78., 79.]],

  1124.                         [[81., 83., 84.],

  1125.                          [86., 88., 89.],

  1126.                          [91., 93., 94.],

  1127.                          [96., 98., 99.]],

  1128.                         [[101., 103., 104.],

  1129.                          [106., 108., 109.],

  1130.                          [111., 113., 114.],

  1131.                          [116., 118., 119.]]]])

  1132.     output = gather(x, indices)

  1133.     error = np.ones(shape=output.asnumpy().shape) * 1.0e-6

  1134.     diff = output.asnumpy() - expect

  1135.     assert np.all(diff < error)

  1136.     assert np.all(-diff < error)