如何访问特定的张量行？
How to access to a specific tensor row?
如何提取张量中的特定行？我有点困惑。
例如，假设我们想要第 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]
另一个想法是重塑数组和 select 第 17 行：
select_row = 17
a.reshape(int(len(a.flatten())/5),5)[select_row-1]
Output: array([8, 1, 6, 3, 4])
如何提取张量中的特定行？我有点困惑。
例如，假设我们想要第 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]
另一个想法是重塑数组和 select 第 17 行：
select_row = 17
a.reshape(int(len(a.flatten())/5),5)[select_row-1]
Output: array([8, 1, 6, 3, 4])