在 Python 中,您可以使用 numpy
库中的 reshape
方法对张量进行重新形状。您可以指定新的形状,以使所有张量具有相同的大小。例如:
import numpy as np
# 定义一个形状为 (3, 4) 的张量
tensor = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
# 重新形状为 (4, 3)
new_tensor = np.reshape(tensor, (4, 3))
print(new_tensor)
# 输出:
# [[ 1 2 3]
# [ 4 5 6]
#