python中怎么改变张量的指定维度

你可以使用NumPy中的transpose函数来改变张量的指定维度。例如，如果你有一个3维张量，你可以使用以下代码将第一维和第二维交换：

import numpy as np

# 创建一个 3x4x5 的张量
tensor = np.random.rand(3, 4, 5)

# 将第一维和第二维交换
tensor_transpose = np.transpose(tensor, (1, 0, 2))

print(tensor.shape)           # 输出 (3, 4, 5)
print(tensor_transpose.shape) # 输出 (4, 3, 5)

在上面的代码中，transpose函数的第二个参数是一个元组，指定新张量中每个维度的位置。在这个例子中，我们将第一维和第二维交换，所以元组的第一个元素是1，第二个元素是0。