pytorch的Tensor的操作
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch的Tensor的操作相关的知识，希望对你有一定的参考价值。
参考技术A 和TensorFlow的tf.concat类似。torch.cat([a , b] , dim),合并tensor a和b，dim指的是从哪个维度。其他维度需要保持一致，如果不一致会出错。
stack 与 concat 不同之处，会增加一个维度用于区分合并的不同 tensor。需要要合并两个 tensor 形状完全一致，而 dim=2 维度前添加一个维度。
类似于numpy中的resize，改变tensor的size。
输出
Tensor.szie()可以获取tensor的形状。
pytorch 合并和拆分
以上是关于pytorch的Tensor的操作的主要内容，如果未能解决你的问题，请参考以下文章
Pytorch | 详解Pytorch科学计算包——Tensor
Pytorch Tensor 维度操作的形象理解 Tensor.unsqueeze() Tensor.squeeze()