torch.cat()函数的官方解释，详解以及例子

可以直接看最下面的例子，再回头看前面的解释，就很明白了。

在pytorch中，常见的拼接函数主要是两个，分别是：

stack()
cat()

一般torch.cat()是为了把多个tensor进行拼接而存在的。实际使用中，和torch.stack()使用场景不同：参考链接torch.stack()，但是本文主要说cat()。

torch.cat() 和python中的内置函数cat()，在使用和目的上，是没有区别的，区别在于前者操作对象是tensor。

1. `cat()`

函数目的：在给定维度上对输入的张量序列seq 进行连接操作。

`outputs = torch.cat(inputs, dim=?) → Tensor`

参数

inputs : 待连接的张量序列，可以是任意相同Tensor类型的python 序列
dim : 选择的扩维, 必须在0到len(inputs[0])之间，沿着此维连接张量序列。

2. 重点

输入数据必须是序列，序列中数据是任意相同的shape的同类型tensor
维度不可以超过输入数据的任一个张量的维度

3.举例子

准备数据，每个的shape都是[2,3]

# x1
x1 = torch.tensor([[11,21,31],[21,31,41]],dtype=torch.int)
x1.shape # torch.Size([2, 3])
# x2
x2 = torch.tensor([[12,22,32],[22,32,42]],dtype=torch.int)
x2.shape  # torch.Size([2, 3])

合成inputs

'inputs为２个形状为[2 , 3]的矩阵 '
inputs = [x1, x2]
print(inputs)
'打印查看'
[tensor([[11, 21, 31],[21, 31, 41]], dtype=torch.int32),tensor([[12, 22, 32],[22, 32, 42]], dtype=torch.int32)]

3.查看结果, 测试不同的dim拼接结果

In    [1]: torch.cat(inputs, dim=0).shape
Out[1]: torch.Size([4,  3])In    [2]: torch.cat(inputs, dim=1).shape
Out[2]: torch.Size([2, 6])In    [3]: torch.cat(inputs, dim=2).shape
IndexError: Dimension out of range (expected to be in range of [-2, 1], but got 2)

大家可以复制代码运行一下就会发现其中规律了。

总结

通常用来，把torch.stack得到tensor进行拼接而存在的。

torch.cat()函数的官方解释，详解以及例子相关推荐

PHP函数call_user_func和call_user_func_array详解
今天在群里面,有个叫lewis的在问call_user_func_array的用法,因为之前一直没有用过,也不能说什么,于是看一下手册,发现是这么写的: call_user_func_array (P ...
php。defined,PHP defined()函数的使用图文详解
PHP defined()函数的使用图文详解 PHP defined() 函数例子定义和用法 defined() 函数检查某常量是否存在. 若常量存在,则返回 true,否则返回 false. 语 ...
rtmp官方协议详解
标准规范学习: rtmp消息结构,包括几个部分: 时戳:4 byte,单位毫秒.超过最大值后会翻转. 长度:消息负载的长度. 类型ID:Type Id 一部分ID范围用于rtmp的控制信令.还有一部 ...
如何使用指向类的成员函数的指针（详解！）
原文:如何使用指向类的成员函数的指针(详解!) 另外一篇英文参考:Member Function Pointers and the Fastest Possible C++ Delegates 我们首 ...
Sklearn中predict_proba函数用法及原理详解
Sklearn中predict_proba函数用法及原理详解(以logistic回归为例) 网上对predict_proba的数学原理解释的太少了,也不明确,特意总结一下,并给出有些不能用该方法的原因 ...
【线性回归 Linear-Regression torch模块实现与源码详解深度学习 Pytorch笔记 B站刘二大人（4/10）】
torch模块实现与源码详解深度学习 Pytorch笔记 B站刘二大人深度学习 Pytorch笔记 B站刘二大人(4/10) 介绍至此开始,深度学习模型构建的预备知识已经完全准备完毕. 从本章开 ...
【学习笔记】薛定谔的喵咪Cat—球盒问题（全详解）
[学习笔记]薛定谔的喵咪Cat-球盒问题(全详解) 传送门:薛定谔的喵咪 \(Luogu-U77460\)(自自上传的题目,数据略水,尤其是 \(opt=9\) ,以后找时间补上) [题目描述] 当一 ...
python scatter参数详解_Python 中 scatter 函数参数及用法详解
Python 中 scatter 函数参数及用法详解 Python 中 scatter 函数参数及用法详解这里有新鲜出炉的 Python 教程, 程序狗速度看过来! Python 编程语言 Pyth ...
python实现排序函数_Python排序函数的使用方法详解
Python排序函数完美体现了Python语言的简洁性,对于List对象,我们可以直接调用sort()函数(这里称为"方法"更合适)来进行排序,而对于其他可迭代对象(如set,di ...

torch.cat()函数的官方解释，详解以及例子

可以直接看最下面的例子，再回头看前面的解释，就很明白了。

1. `cat()`

`outputs = torch.cat(inputs, dim=?) → Tensor`

2. 重点

3.举例子

总结

torch.cat()函数的官方解释，详解以及例子相关推荐

最新文章

热门文章

torch.cat()函数的官方解释，详解以及例子

可以直接看最下面的例子，再回头看前面的解释，就很明白了。

1. cat()

outputs = torch.cat(inputs, dim=?) → Tensor

2. 重点

3.举例子

总结

torch.cat()函数的官方解释，详解以及例子相关推荐

最新文章

热门文章

1. `cat()`

`outputs = torch.cat(inputs, dim=?) → Tensor`