如何访问特定的张量行？

问题描述

如何提取张量中的特定行？我有点困惑。

例如，假设我们要有第17行 specified row

import numpy as np
np.random.seed(seed = 5)
a = np.random.randint(10,size=(2,3,4,5))
a

解决方法

在这种情况下，您可以执行以下操作：

a[1][1][0]

或a[1,1,0]

另一个想法是重塑数组并选择第17行：

select_row = 17
a.reshape(int(len(a.flatten())/5),5)[select_row-1]

输出：array（[8，1，6，3，4]）