如果我们想将一个形状为(32, 96) 的图像张量变形成形状为 (32, 32, 3) (32, 96) 的张量，代码

如下：

import numpy as np

# 创建一个形状为 (32, 96) 的图像张量
image_tensor = np.zeros((32, 96))

# 将图像张量变形为 (32, 32, 3) 的张量
new_shape = (32, 32, 3)
new_tensor = np.zeros(new_shape)
for i in range(new_shape[0]):
    for j in range(new_shape[1]):
        new_tensor[i, j, :] = image_tensor[i, j*3:(j+1)*3]

print(new_tensor.shape)

注：这里假设原始图像的每行包含了 3 个颜色通道的值。如果原始图像每行只包含一个颜色通道的值，可以将内层循环改成 `new_tensor[i, j, :] = [image_tensor[i, j], 0, 0]`。