stack() with better dtype promotion rules.
(arrays, axis=0)
| 454 | |
| 455 | |
| 456 | def stack(arrays, axis=0): |
| 457 | """stack() with better dtype promotion rules.""" |
| 458 | xp = get_array_namespace(arrays[0]) |
| 459 | return xp.stack(as_shared_dtype(arrays, xp=xp), axis=axis) |
| 460 | |
| 461 | |
| 462 | def reshape(array, shape): |
searching dependent graphs…