快速从b中取出a的形状的向量(利用Slice)
前置知识
首先理解Python中切片的使用就能明白slice了
#例如
a = [1, 2, 3, 4, 5, 6,7]
a[3:5]
>>> [4, 5]
# 其等价于
a[3:5:1]
>>> [4, 5]
因为本质上平时省略了步数,默认为1
所以一个Slice(None)print出来时是:
(none,none,none)
所以可以快速写出如下代码:
num_dimensions_to_slice = len(a.shape)
indices = [slice(None)] * num_dimensions_to_slice
for dim in range(num_dimensions_to_slice):
indices[dim] = slice(a.shape[dim])
a=store[key][-1][indices]