7 lines
175 B
Python
7 lines
175 B
Python
|
import cupy
|
||
|
|
a = cupy.asarray([[[1, 2, 3],[4,5,6],[7,8,9]],[[10, 11, 12],[13,14,15],[16,17,18]]])
|
||
|
|
print(a)
|
||
|
|
a= cupy.reshape(a,(-1,3))
|
||
|
|
print(a)
|
||
|
|
a = cupy.median(a,axis=0)
|
||
|
|
print(a)
|