【Pytorch】expand()用法==》扩展某个维度

有两点需要注意，无论是 expand() 还是 expand_as()：

只能在第0维扩展一个维数，比如原来是是（1,3,4）==》（2,1,3,4），而在其他维度扩展不可以（1,3,4）==》（1,2,3,4）【错误】
如果不增加维数，只是增加维度，要增加的原维度必须是1才可以在该维度增加维度，其他值均不可以

import torch
#1
x = torch.randn(2, 1, 1)#为1可以扩展为3和4
x = x.expand(2, 3, 4)
print('x :', x.size())
>>> x : torch.Size([2, 3, 4])
#2
#扩展一个新的维度必须在最前面，否则会报错
x = x.expand(2, 3, 4, 6)
>>> RuntimeError: The expanded size of the tensor (3) must match the existing size (2) at non-singleton dimension 1.
x = x.expand(6, 2, 3, 4)
>>> x : torch.Size([6, 2, 3, 4])
#3
#某一个维度为-1表示不改变该维度的大小
x = x.expand(6, -1, -1, -1)
>>> x : torch.Size([6, 2, 1, 1])

import torch
#1
x = torch.randn(2, 1, 1)#原维度为1可以扩展为其他维度
y = torch.randn(2, 3, 3)
x = x.expand_as(y)
print('x :', x.size())
>>> x : torch.Size([2, 3, 3])
#2
x = torch.randn(2, 2, 2)#原维度为其他不是1的值不可以扩展为其他维度
y = torch.randn(2, 3, 4)
x = x.expand_as(y)
print('x :', x.size())
>>> RuntimeError: The expanded size of the tensor (4) must match the existing size (2) at non-singleton dimension 2.  Target sizes: [2, 3, 4].

【Pytorch】expand()用法==》扩展某个维度相关推荐

【PyTorch】 tensor.squeeze压缩维度
[PyTorch] tensor.squeeze压缩维度 1.背景 2.squeeze方法 3.示例 1.背景 pytorch的tensor维度扩展.压缩是一个常见的操作, tensor维度扩展参见如 ...
机器学习之神经网络（从神经网络结构到pytorch基本用法）
机器学习之神经网络(从神经网络结构到pytorch基本用法) 1.神经网络概念 1.1 网络结构人工神经网络(artificial neural network,ANN),简称神经网络(neur ...
Pytorch基本用法乘法函数详解：5个张量乘法function
1.基础张量维度:维度个数和维度大小:.ndim可查看维度个数,.shape可查看维度大小. 如下代码,张量a:维度个数为2,是一个2维张量:维度大小为[2,3],即第0维的维度大小为2,第1维为3 ...
pytorch中的torch.tensor.repeat以及torch.tensor.expand用法
文章目录 torch.tensor.expand torch.tensor.repeat torch.tensor.expand 先看招 import torch x = torch.tensor([ ...
PyTorch expand() 函数
文章目录 1.官方解释 2.例子 2.1.扩展到更大维度 2.2.添加新的维度 1.官方解释 Tensor.expand(*sizes) -> Tensor:对一个张量进行扩张返回的是一个 t ...
pytorch CrossEntropyLoss用法
数据类型只支持long类型只支持两个维度: import torch import torch.nn as nnloss = nn.CrossEntropyLoss()# input, NxC=2x ...
pytorch 笔记：扩展torch.autograd
1 扩展torch.autograd 向 autograd 添加操作需要为每个操作实现一个新的 Function 子类. 回想一下,函数是 autograd 用来编码操作历史和计算梯度的东西. 2 何 ...
这可能是关于Pytorch底层算子扩展最详细的总结了！
1.前言一般情况下,pytorch推荐使用python层的前端语言来构建新的算子.因为pytorch在python层的api已经足够丰富,可以构造出很多自定义的算子.但是有时候出于一些其他方面的 ...
Npm配置全局安装插件目录以及命令用法扩展
1 配置全局插件文件存放目录说明:本地nodejs安装路径为:D:\Software\Nodejs 先在安装目录创建2个空的文件夹 cd /d D:\Software\Nodejs mkdir no ...

【Pytorch】expand()用法==》扩展某个维度

【Pytorch】expand()用法==》扩展某个维度相关推荐

最新文章

热门文章