Pytorch学习(一)(读代码中不懂的知识点)

最新推荐文章于 2022-09-23 11:34:46 发布

猪小肥呀

最新推荐文章于 2022-09-23 11:34:46 发布

阅读量137

点赞数

文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/fighterlucky/article/details/124130667

版权

本文介绍了PyTorch中的基本张量操作，包括使用torch.linspace创建等差序列，以及torch.cat函数在维度0和1上的拼接用法。torch.unsqueeze用于增加一维，而torch.squeeze则用于移除单个维度的一维。这些概念对于理解和操作PyTorch张量至关重要。

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

torch.linspace(start, end, steps=100, out=None) → Tensor: 返回一个1维张量，包含在区间start和end上均匀间隔的step个点。
C = torch.cat( (A,B),0 ) 按维数0拼接（竖着拼/上下拼）
C = torch.cat( (A,B),1 ) 按维数1拼接（横着拼/左右拼）（-1 与 1相同）
torch.unsqueeze() 和 torch.squeeze()

x=torch.Tensor([1, 2, 3, 4]) #（tensor([1., 2., 3., 4.])）
torch.unsqueeze(x,1) #

在这里插入图片描述

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

关注关注