Matlab 堆叠张量
有时候我们需要堆叠多个 Tensor
,例如重建了
N
N
N 张
74
×
66
×
70
74\times66\times70
74×66×70的磁共振图像,我们希望打包成
N
×
74
×
66
×
70
N \times 74 \times 66\times70
N×74×66×70用于深度学习相关任务。
对于熟悉 numpy
和 PyTorch
的用户来说,这个流程分为两步,首先将每个图像使用 unsqueeze()
拓展出一个大小为
1
1
1 的维度,然后使用 stack()
在此维上进行堆叠即可。但是 Matlab
中并没有拓展一维的相关功能。
事实上,Matlab
支持这样的扩维,即 data_stack(0, :, :, : ) = data(:, :, :)
直接写明要拓展的维度进行赋值即可,无需额外的扩维操作。