假设我有一个多维的numpy数组A。我正在尝试获取我选择的任意维度的A切片，例如如果A是3D数组，并且给出了一些整数layer和某些尺寸dim：
A
layer
dim
如果dim = 2，则选择A [：，：，layer]
如果dim = 1，请选择A [：，layer，：]
如果dim = 0，则选择A [layer，：，：]
对于我来说，A可能有很多尺寸可供选择，所以我不想手动编写。什么是简单易读的方法？