cupy.compress#

cupy.compress(condition, a, axis=None, out=None)[源]#

返回沿给定轴选择的数组切片。

参数:

  • condition (一维布尔数组) – 选择要返回哪些条目的数组。如果 len(condition) 小于沿给定轴的数组 a 的大小，则输出会被截断为 condition 数组的长度。

  • a (cupy.ndarray) – 从中提取部分的数组。

  • axis (int) – 沿其进行切片的轴。如果为 None（默认），则对展平的数组进行操作。

  • out (cupy.ndarray) – 输出数组。如果提供，应具有适当的形状和数据类型。

返回值:

删除 condition 为 false 的沿轴切片后的数组 a 的副本。

返回类型:

cupy.ndarray

警告

此函数可能会同步设备。

另请参阅

numpy.compress()