pytorch中tensor剪切行或列

使用示例

x = x[:,:,torch.arange(x.size(2))%8!=0]
剪切掉第三个维度的8的整数行
维度上不仅支持用:进行多行选择,也可以用bool组成的tensor判断是否保存这一行。

posted @ 2023-10-12 21:13  心比天高xzh  阅读(55)  评论(0)    收藏  举报