Pytorch 实践中学习笔记:
程序员文章站
2024-02-27 13:49:45
...
Pytoch 学习
对于pytorch,tensorflow 这类计算工具,我认为,花大量的时间看文档、时间去学习的效率是十分低效的:
- 很多知识看过之后不去使用,基本上都会忘记
- 另外在实践中学习,能够让知识学习地更加牢固,也更加高效
- 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])