torch中的split函数学习

说文科技

于 2021-11-13 11:23:01 发布

阅读量1.4k

点赞数

CC 4.0 BY-SA版权

分类专栏： # pytorch 文章标签： pytorch 深度学习机器学习

喜欢文章？请私信联系作者。

53 篇文章

订阅专栏

这篇博客介绍了Python深度学习库PyTorch中的split函数。通过示例代码展示了如何沿着指定维度将张量切分成多个部分，帮助理解该函数在处理多维数据时的作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

split函数的学习

import torch
weight = torch.randn(3,2,2)
print(weight)
a = torch.split(weight, 1, dim=0) # 沿着dim这个维度切分，然后每个tensor的切割单位都是1。
print(a)

在这里插入图片描述