CuPy
v7.8.0
Overview
Tutorial
Reference Manual
Multi-Dimensional Array (ndarray)
Universal Functions (ufunc)
Routines
Array Creation Routines
Array Manipulation Routines
Binary Operations
Data Type Routines
FFT Functions
Indexing Routines
cupy.c_
cupy.r_
cupy.nonzero
cupy.where
cupy.indices
cupy.ix_
cupy.unravel_index
cupy.take
cupy.take_along_axis
cupy.choose
cupy.diag
cupy.diagonal
cupy.lib.stride_tricks.as_strided
cupy.place
cupy.put
cupy.fill_diagonal
Input and Output
Linear Algebra
Logic Functions
Mathematical Functions
Padding
Random Sampling (
cupy.random
)
Sorting, Searching, and Counting
Statistical Functions
CuPy-specific Functions
SciPy-compatible Routines
Sparse matrices
Multi-dimensional image processing
NumPy-CuPy Generic Code Support
Memory Management
Low-Level CUDA Support
Kernel binary memoization
Custom kernels
Interoperability
Testing Modules
Profiling
Environment variables
Difference between CuPy and NumPy
Comparison Table
Development
API Compatibility Policy
Contribution Guide
Misc Notes
Installation Guide
[Experimental] Installation Guide for ROCm environemt
Upgrade Guide
License
CuPy
Docs
»
Reference Manual
»
Routines
»
Indexing Routines
»
cupy.choose
Edit on GitHub
cupy.choose
¶
cupy.
choose
(
a
,
choices
,
out=None
,
mode='raise'
)
[source]
¶