tensor 增加一维度_Pytorch之Tensor操作

最新推荐文章于 2024-07-24 19:41:07 发布

兔纸周嬷嬷

最新推荐文章于 2024-07-24 19:41:07 发布

阅读量7.2k

点赞数 1

文章标签： tensor 增加一维度

本文链接：https://blog.csdn.net/weixin_33001561/article/details/112536119

版权

本文介绍了Pytorch中用于处理Tensor维度的几种方法，包括flatten、reshape、squeeze、unsqueeze、cat、view和transpose。重点讨论了reshape和view函数的区别，强调了它们在返回视图和复制数据方面的不同行为，以及何时可以安全地进行不复制的数据视图操作。

摘要由CSDN通过智能技术生成

1. torch.flatten(input,start_dim=0,end_dim=-1)

2.torch.reshape(input,shape)

3.torch.squeeze(input, dim=None, out=None)

4.torch.unsqueeze(input, dim)

5.torch.cat(tensors, dim=0, out=None)

6.view(*shape)

7.torch.transpose(input, dim0, dim1)

下面记录一下reshape和view函数的区别：

reshape()：Returns a tensor with the same data and number of elements as input, but with the specified shape. When possible, the returned tensor will be a view of input. Otherwise, it will be a copy. Contiguous inputs and inputs with compatible strides can be reshaped without copying, but you should not depend on the copying vs. viewing behavior.See torch.Tensor.view()