Closed
Description
Need to find a TF equivalent API in executing the JAX ops.index_update functionality.
utils.py", line 410, in get_masked_array
raise TypeError(x, type(x))
TypeError: (DeviceArray([[ 0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ],
[ 0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ,
0. , 0. , 0. , 0. ],
[ 0.47055304, 0.6285802 , -0.22192092, -0.667655 ,
0.10804348, -0.2439548 , 0.37487373, -1.5117857 ,
0.13458425, -0.34092295, 0.34964558, 0.4743455 ,
-0.5111295 , 0.08342728, -0.10011845, -0.42951363,
0.12291951, -2.3555408 , 0.06692982, -0.2854176 ,
1.0401559 , 0.97515863, -1.4384125 , 1.2863765 ,
0.08358394, -0.7866839 , 0.23288795, -1.4278252 ,
-0.11413796, -0.70401907, 2.2100668 , -1.5638859 ,
0.47637904, -0.19487365, -0.99257374, 0.7704442 ,
-0.30862156, -0.90175354, -1.1160144 , 0.7503657 ,
-0.6176687 , 0.3577367 , -1.0693784 , -0.5346328 ,
-0.52183443, 1.5098324 , -1.0317351 , -1.8861948 ,
2.4031775 , -1.1306083 , 1.0992835 , 0.21266708,
0.5999638 , -1.9048934 , -0.6604947 , -0.6230653 ,
-0.06880049, -1.2365948 , 0.59906626, 0.789022 ,
-0.18167678, -0.37683064, 0.9781549 , 1.23865 ,
0.2011511 , 0.42039153, 0.10758805, 0.29374722,
-0.62673444, -0.27408686, 0.9471005 , -0.01523335,
-0.01991809, -0.15252574, -1.4507486 , 1.6987492 ,
-1.2424468 , -0.4143933 , 0.5367061 , -0.18643558,
-1.64998 , -0.3848491 , -0.88372403, -1.2557826 ,
1.3627311 , -1.019953 , -0.57100815, 0.68098336,
0.46940225, 0.28183317, 2.0002983 , -0.9182061 ,
-1.7851701 , -1.0026677 , 0.3424797 , -0.26637673,
-0.20781823, -1.7653099 , -0.055797 , 0.42224863,
1.4412493 , 0.20655143, 0.26329288, -0.26830837,
-0.6690551 , -0.8026767 , -1.6743355 , 0.48559475,
-0.12912928, 1.1803355 , -0.84874505, -1.2203509 ,
0.5168695 , 0.4992112 , -0.8779885 , -0.3857809 ,
0.05366755, -0.3525355 , 0.14090906, 1.0464299 ,
0.7789615 , 0.79876614, -1.1611547 , -0.01883734,
0.57565856, -1.6148225 , -1.7661449 , 1.1159137 ],
[ 1.7436517 , 1.2403702 , -0.6869173 , 0.9129569 ,
0.23201469, -0.6145854 , 1.9160763 , -1.653898 ,
-1.3471946 , -1.8016372 , 0.05446442, 0.45075658,
-1.1183583 , -0.99564385, -1.1773059 , -0.816847 ,
1.431208 , -0.13749348, 0.12738436, 0.32845214,
-0.01712162, -0.5692807 , -0.56701005, 0.08576272,
-0.60248464, 1.1343902 , -1.2910378 , -1.3916879 ,
-0.20836708, 1.2733467 , 0.83153296, 1.3242177 ,
0.5120684 , -0.24033947, 0.91572773, 2.270148 ,
0.69143736, -1.8877497 , -0.16107106, -2.3108761 ,
0.14434606, -0.6666905 , -0.7064969 , -1.6592307 ,
1.0491127 , 0.9672468 , 0.93105197, -0.8245607 ,
-1.6271671 , -0.8980524 , 0.04044109, -1.5875283 ,
0.481785 , 0.2217058 , 0.01285424, 1.2344213 ,
1.9342787 , -0.46035308, 1.1991669 , 1.6305301 ,
-0.59450793, -1.1630133 , -1.4455531 , -1.5925426 ,
-1.9613711 , 1.7310132 , 0.9979003 , 0.20509146,
0.91236615, 0.6276525 , -0.08466922, -0.41186643,
1.3045098 , -0.44807404, 0.24899247, -0.49574265,
0.89383435, 1.822114 , 0.5317234 , -0.42532927,
-1.6176043 , -0.28715384, -0.9871549 , 0.62340117,
0.8301923 , 0.4759204 , 0.8530764 , 0.13110952,
-0.38294357, -0.53054744, 0.8554194 , 0.84938085,
0.4945622 , 0.935319 , 1.2166226 , 0.36939874,
-0.46802616, 1.8939401 , -1.7973521 , 0.06464902,
0.4062858 , -0.5966817 , -1.5532738 , 0.47134528,
-0.8542637 , -0.25430223, 0.99411577, -0.49718758,
0.24050687, 0.02138576, 0.3947097 , 1.3843871 ,
1.0715213 , 0.8659403 , 0.7066823 , 0.14143308,
1.105173 , 0.19036756, 0.4168512 , -1.1783463 ,
-0.24360779, 1.3549082 , 1.0244592 , 0.09035202,
-0.9024119 , -1.2422194 , -0.09328686, -1.1991651 ]], dtype=float32), <class 'jax.interpreters.xla.DeviceArray'>)