当前位置: 首页 > 工具软件 > Chunk Scatter > 使用案例 >

pytorch中split,chunk,unblind和scatter使用

章稳
2023-12-01

关于split和chunk使用与区别:
split和chunk

在对比学习代码中,对特征变换使用split和cat组合,可以快速简单调整维度;

contrast_feature = torch.cat(torch.unbind(features, dim=1), dim=0)

unbind使用

pytorch-scatter使用
scatter 可以理解成放置元素或者修改元素

 类似资料: