Matlab 堆叠张量

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用于深度学习相关任务。
对于熟悉 numpyPyTorch 的用户来说,这个流程分为两步,首先将每个图像使用 unsqueeze() 拓展出一个大小为 $1 $ 的维度,然后使用 stack()在改维上进行堆叠即可。但是 Matlab 中并没有拓展一维的相关功能。
事实上,Matlab 支持这样的扩维,即 data_stack(0, :, :, : ) = data(:, :, :)直接写明要拓展的维度进行赋值即可,无需额外的扩维操作。

你可能感兴趣的:(杂记,matlab)