pytorch学习记录

Posted yanxingang

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch学习记录相关的知识,希望对你有一定的参考价值。

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
= 5 select_channel = random.sample(channel_list,k) select_channel = torch.tensor(select_channel,dtype=torch.long)
# 从小到大排下序 select_channel,i
= torch.sort(select_channel) print(select_channel) select_tensor = torch.index_select(data,1,select_channel)

2、转置图像和翻转图像,平移图像

转置直接用transpose即可,后面只跟两个参数,就是两个要转置的维度。

以上是关于pytorch学习记录的主要内容,如果未能解决你的问题,请参考以下文章

PyTorch深度学习:用PyTorch实现线性回归

深度学习——基础(基于Pytorch代码)

PyTorch学习问题记录

pytorch学习记录

深度学习Pytorch框架的入门简易代码模板及解析

ElasticSearch学习问题记录——Invalid shift value in prefixCoded bytes (is encoded value really an INT?)(代码片段