![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 92
冲冲冲鸭鸭鸭~
这个作者很懒,什么都没留下…
展开
-
张量(Tensor)操作
1、torch.cat():将张量按照dim维度进行拼接2、torch.stack():将张量在新创建的dim维度上进行拼接3、torch.chunk():将张量按照维度 dim 进行平均切分。若不能整除,则最后一份张量小于其他张量4、torch.split():将张量按照维度 dim 进行平均切分。可以指定每一个分量的切分长度5、torch.index_select():在维度 dim 上,按照 index 索引取出数据拼接为张量返回原创 2022-09-11 15:40:59 · 2627 阅读 · 0 评论 -
Tensor的创建方法
1、直接创建Tensor——torch.tensor()2、从numpy创建tensor——torch.from_numpy(ndarray)3、根据数值创建tensor原创 2022-09-05 20:47:21 · 11941 阅读 · 0 评论