首页 > 解决方案 > 沿特定维度洗牌张量

问题描述

我有一个 4D 张量 [batch_size, temporal_dimension, data[0], data[1]],[temporal_dimension, data[0], data[1]] 的 3d 张量实际上是我对网络的输入数据。我会沿着第二维度打乱张量,这是我的时间维度,以检查网络是否正在从时间维度学习一些东西。如果这种改组是可重现的,我会很高兴。

有没有某种方法可以做到这一点?

谢谢。

标签: pythonpytorchshuffletensor

解决方案


推荐阅读