pytorch 中 expand （）函数

expand函数的功能就是用来扩展张量中某维数据的尺寸，它返回输入张量在某维扩展为更大尺寸后的张量。
例如：

x = torch.tensor([1, 2, 3])
y = x.expand(2, 3)
y1 = x.expand(3,3)
print(x.size())
print(x)
print(y)
print(y1)

输出：

torch.Size([3])
tensor([1, 2, 3])
tensor([[1, 2, 3],[1, 2, 3]])
tensor([[1, 2, 3],[1, 2, 3],[1, 2, 3]])

由此可以看到expand扩展维度会复制原有的数据进行扩展，我们以下面的例子进行验证。

x = torch.tensor([[1], [2], [3]])
y = x.expand(3, 3)
y1 = x.expand(4,3)
print(x.size())
print(y)
print(y1)

输出：

torch.Size([3, 1])
tensor([[1, 1, 1],[2, 2, 2],[3, 3, 3]])
tensor([[1, 1, 1],[2, 2, 2],[3, 3, 3]])

pytorch 中 expand （）函数相关推荐

pytorch 中 contiguous() 函数理解
pytorch 中 contiguous() 函数理解文章目录 pytorch 中 contiguous() 函数理解引言使用 contiguous() 后记文章抄自 Pytorch中cont ...
pytorch中repeat()函数理解
pytorch中repeat()函数理解最近在学习过程中遇到了repeat()函数的使用,这里记录一下自己对这个函数的理解. 情况1:repeat参数个数与tensor维数一致时 a = torch ...
**Pytorch 中view函数和reshape函数的区别*
Pytorch 中view函数和reshape函数的区别(我是一名大一刚学计算机的学生希望我的说法对你有帮助) 首先:要了解这个问题我们要先了解一个基本知识张量的储存方式跟据图片我们可以清楚的看 ...
Pytorch中apply函数作用
pytorch中的model.apply(fn)会递归地将函数fn应用到父模块的每个子模块submodule,也包括model这个父模块自身.经常用于初始化init_weights的操作.如下appl ...
matlab expand 扩展用不了,matlab中expand函数 Matlab 里的expand怎么用
Matlab 里的expand怎么用 expand函数用于多项式的展开运算, syms x y;%定义x,y变量 expand((x-2)*(x-4)) ans = x^2 - 6*x + 8今天小编 ...
python batchnorm2d_BatchNorm2d原理、作用及其pytorch中BatchNorm2d函数的参数讲解
BN原理.作用: 函数参数讲解: BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) 1. ...
pytorch中（函数名有下划线和没有下划线）clamp和clamp_的区别
在pytorch中,我们需要注意,对于tensor的操作时候是in-place类型. in-place类型是指,但在一个tensor上操作了之后,是直接修改了这个tensor,还是返回一个新的tens ...
Pytorch中contiguous()函数理解
引言在pytorch中,只有很少几个操作是不改变tensor的内容本身,而只是重新定义下标与元素的对应关系的.换句话说,这种操作不进行数据拷贝和数据的改变,变的是元数据. 会改变元数据的操作是: n ...
Pytorch丨expand函数
1.官方文档描述 expand(*sizes) 返回tensor的一个新视图,单个维度扩大为更大的尺寸. tensor也可以扩大为更高维,新增加的维度将附在前面. 扩大tensor不需要分配新内存,只 ...

pytorch 中 expand （）函数

pytorch 中 expand （）函数

pytorch 中 expand （）函数相关推荐

最新文章

热门文章