2019年3月9日
pytorch学习记录
摘要： 1、pytorch中的torch.split只能将tensor分割为相等的几分，如果需要特定的需求将tensor分割开，可以用torch.index_select。使用的时候，先生成index索引，示例程序如下： 2、转置图像和翻转图像，平移图像 转置直接用transpose即可，后面只跟两个参数， 阅读全文
posted @ 2019-03-09 15:15 阿刚的代码进阶之旅 阅读(159) 评论(0) 推荐(0) 编辑