1.pytorch中的torch.split只能将tensor分割为相等的几分,如果需要特定的需求将tensor分割开,可以用torch.index_select.使用的时候,先生成index索引,示例程序如下: import torch import random# tensor data = torch.randn(10,10,10,10) channel_list = [i for i in range(0,data.shape[1])]# 在channel_list中返回k个随机数 k…