python获得张量的行列数

在Python中，可以使用NumPy库来处理张量。如果想要获取张量的行和列数，可以使用NumPy中的shape属性。假设我们有一个2x3的张量，可以按照以下方式获取它的行和列数：

import numpy as np

tensor = np.array([[1, 2, 3], [4, 5, 6]])
rows, cols = tensor.shape
print("Rows:", rows)
print("Cols:", cols)

输出：

Rows: 2
Cols: 3

这里我们使用NumPy的array函数创建了一个2x3的张量，并使用shape属性获取了它的行和列数。rows变量的值为2，cols变量的值为3。