欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Pytorch 实践中学习笔记:

程序员文章站 2024-02-27 13:49:45
...

Pytoch 学习

对于pytorch,tensorflow 这类计算工具,我认为,花大量的时间看文档、时间去学习的效率是十分低效的:

  1. 很多知识看过之后不去使用,基本上都会忘记
  2. 另外在实践中学习,能够让知识学习地更加牢固,也更加高效
  • permute(dims) :将tensor的维度换位。
    参数:参数是一系列的整数,代表原来张量的维度。比如三维就有0,1,2这些dimension。
    例:
import torch
import numpy as np
a=np.array([[[1,2,3],[4,5,6]]])
unpermuted=torch.tensor(a)
print(unpermuted.size())  #  ——>  torch.Size([1, 2, 3])
permuted=unpermuted.permute(2,0,1)
print(permuted.size())     #  ——>  torch.Size([3, 1, 2])
相关标签: 算法工程师笔记