g
y
7
7
7
7
2020年10月7日
Pytorch-tensor的分割，属性统计
摘要： 1.矩阵的分割 方法：split(分割长度,所分割的维度)，split([分割所占的百分比],所分割的维度) a=torch.rand(32,8) aa,bb=a.split(16,dim=0) print(aa.shape) print(bb.shape) cc,dd=a.split([20,12 阅读全文
posted @ 2020-10-07 21:48 gy77 阅读(1010) 评论(0) 推荐(0) 编辑