PyTorch 中如何反转矩阵的行顺序 (z2 矩阵行倒序变换)

可以使用 torch.flip() 函数将 z2 矩阵按照行倒叙变换。

具体实现如下：

z2 = torch.flip(z2, dims=[0])

将 z2 矩阵按照行倒叙变换后，再将其代入 loss_fn() 函数计算损失。

原文地址: https://www.cveoy.top/t/topic/lNuH 著作权归作者所有。请勿转载和采集!