shadow_tensor的维度是{16,19,64,64},将其扩展为{16,19,1,64,64}
时间: 2023-03-26 08:02:36 浏览: 74
pytorch改变tensor维度
可以使用numpy中的reshape函数将shadow_tensor的维度扩展为{16,19,1,64,64},代码如下:
import numpy as np
shadow_tensor = np.zeros((16,19,64,64))
shadow_tensor = shadow_tensor.reshape((16,19,1,64,64))
注意,这里的reshape函数中的参数是一个元组,表示将原来的维度扩展为{16,19,1,64,64}。
阅读全文