2023年10月12日
pytorch中tensor剪切行或列
摘要： 使用示例 x = x[:,:,torch.arange(x.size(2))%8!=0] 剪切掉第三个维度的8的整数行 维度上不仅支持用：进行多行选择，也可以用bool组成的tensor判断是否保存这一行。 阅读全文
posted @ 2023-10-12 21:13 心比天高xzh 阅读(55) 评论(0) 推荐(0)