torch.tensor().permute(2,1,0)

参考：
permute函数
python pytorch permute函数
[Python] 维度交换函数：transpose(m,n,r)和permute(m,n,r)

import torch
from torch import nn
import numpy as np
b=np.array([[[ 0, 1, 2,3],[ 4,5,6,7],[ 8,9, 10,11]],[[12, 13, 14, 15],[16, 17,18, 19],[20, 21,22, 23]]])
print(b)

[[[ 0  1  2  3][ 4  5  6  7][ 8  9 10 11]][[12 13 14 15][16 17 18 19][20 21 22 23]]]

unpermuted=torch.tensor(b)
permuted = unpermuted.permute(2,1,0)
print(permuted)

tensor([[[ 0, 12],[ 4, 16],[ 8, 20]],[[ 1, 13],[ 5, 17],[ 9, 21]],[[ 2, 14],[ 6, 18],[10, 22]],[[ 3, 15],[ 7, 19],[11, 23]]], dtype=torch.int32)

torch.tensor().permute(2,1,0)相关推荐

PyTorch 1.0 中文文档：torch.Tensor
译者:hijkzzz torch.Tensor 是一种包含单一数据类型元素的多维矩阵. Torch定义了八种CPU张量类型和八种GPU张量类型: Data type dtype CPU tensor ...
pytorch 定义torch类型数据_PyTorch官方中文文档：torch.Tensor
torch.Tensor torch.Tensor是一种包含单一数据类型元素的多维矩阵. Torch定义了七种CPU tensor类型和八种GPU tensor类型: Data tyoe CPU te ...
PyTorch torch.Tensor.contiguous() 用法与理解
中文文档: contiguous() → Tensor 返回一个内存连续的有相同数据的 tensor,如果原 tensor 内存连续则返回原 tensor 英文文档: contiguo ...
PyTorch 笔记（02）— 常用创建 Tensor 方法（torch.Tensor、ones、zeros、eye、arange、linspace、rand、randn、new）
1. Tensor 概念分类 PyTorch 中的张量(Tensor)类似 NumPy 中的 ndarrays,之所以称之为 Tensor 的另一个原因是它可以运行在 GPU 中,以加速运算. 1.1 ...
pytorch中的torch.tensor.repeat以及torch.tensor.expand用法
文章目录 torch.tensor.expand torch.tensor.repeat torch.tensor.expand 先看招 import torch x = torch.tensor([ ...
通俗讲解Pytorch梯度的相关问题：计算图、torch.no_grad、zero_grad、detach和backward；Variable、Parameter和torch.tensor
文章目录 with torch.no_grad()和requires_grad backward() Variable,Parameter和torch.tensor() zero_grad() 计算图 ...
Unable to get repr for＜class‘torch.Tensor‘＞
Unable to get repr for <class 'torch.Tensor'> tensor越界访问后就会变成这样. import torcha_data=torch.Tens ...
unable to get repr for class ‘torch.tensor‘
unable to get repr for class 'torch.tensor' 出错代码: batch_conf.gather(1, conf_t.view(-1,1)) 最近码代码使用pyt ...
torch tensor用法
1.x = torch.Tensor(5):zero() Tensor转int: int(x1.cpu().data) range与arange区别: if __name__ == '__main__ ...

torch.tensor().permute(2,1,0)

torch.tensor().permute(2,1,0)相关推荐

最新文章

热门文章