torch.split()

最新推荐文章于 2022-08-15 11:51:19 发布

懒羊羊001

最新推荐文章于 2022-08-15 11:51:19 发布

阅读量1k

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/freedomsm/article/details/122553996

版权

'''
import torch 
#关于torch.split()函数的使用
ysh=torch.rand((4,8,6))
print('ysh_shape',ysh.shape)
mess=torch.split(ysh,2,dim=0)
print('ysh',ysh)
for i in  mess:
    print("i",i)
#torch.split(tensor,split_size_or_section,dim)完成了在维度dim上对张量tensor切分的任务，切分的大下
#由split_size_or_section决定
演示结果
ysh_shape torch.Size([4, 8, 6])
ysh tensor([[[0.6902, 0.7304, 0.6721, 0.7084, 0.6144, 0.9573],
         [0.6282, 0.1948, 0.4987, 0.2209, 0.5220, 0.5916],    
         [0.0926, 0.5765, 0.1929, 0.8219, 0.0533, 0.2595],    
         [0.0118, 0.6234, 0.9052, 0.9021, 0.5443, 0.9826],    
         [0.3294, 0.8859, 0.2642, 0.4821, 0.2936, 0.7825],    
         [0.0916, 0.6130, 0.3258, 0.7555, 0.2906, 0.3834],    
         [0.9492, 0.9900, 0.3934, 0.6345, 0.3282, 0.3203],    
         [0.4193, 0.3491, 0.0120, 0.3940, 0.0949, 0.2586]],   

        [[0.3945, 0.6685, 0.7638, 0.4580, 0.4213, 0.7744],    
         [0.4684, 0.1031, 0.7223, 0.3876, 0.9374, 0.6348],    
         [0.5073, 0.0811, 0.9718, 0.3441, 0.9789, 0.5309],    
         [0.0464, 0.1444, 0.0131, 0.1354, 0.1363, 0.4884],    
         [0.5902, 0.5752, 0.5572, 0.4773, 0.1481, 0.4861],    
         [0.4762, 0.7723, 0.5584, 0.3330, 0.9923, 0.3006],    
         [0.6967, 0.6267, 0.0114, 0.7994, 0.4867, 0.3571],    
         [0.3426, 0.3432, 0.9019, 0.3500, 0.2407, 0.7816]],

        [[0.3526, 0.6836, 0.5706, 0.5813, 0.2053, 0.8642],
         [0.4879, 0.3136, 0.8652, 0.7691, 0.5845, 0.8990],
         [0.4399, 0.8685, 0.5658, 0.3264, 0.8709, 0.9613],
         [0.0199, 0.5939, 0.9009, 0.9864, 0.6581, 0.1774],
         [0.4650, 0.2194, 0.5409, 0.1912, 0.4927, 0.9850],
         [0.0033, 0.4131, 0.5793, 0.2638, 0.8113, 0.3437],
         [0.8758, 0.4926, 0.2402, 0.8811, 0.6346, 0.3666],
         [0.4559, 0.6895, 0.2985, 0.8002, 0.9424, 0.0689]],

        [[0.1367, 0.3218, 0.4103, 0.9068, 0.1289, 0.4846],
         [0.0283, 0.4955, 0.3459, 0.5642, 0.7386, 0.5170],
         [0.4937, 0.4622, 0.2706, 0.4797, 0.9714, 0.5258],
         [0.5382, 0.6232, 0.8357, 0.2047, 0.2334, 0.5084],
         [0.0239, 0.3009, 0.2565, 0.4092, 0.3953, 0.5668],
         [0.3464, 0.1299, 0.7938, 0.9157, 0.7993, 0.3227],
         [0.7788, 0.2640, 0.0140, 0.1054, 0.1661, 0.2205],
         [0.4375, 0.0124, 0.2400, 0.9916, 0.7229, 0.1512]]])
i tensor([[[0.6902, 0.7304, 0.6721, 0.7084, 0.6144, 0.9573],
         [0.6282, 0.1948, 0.4987, 0.2209, 0.5220, 0.5916],
         [0.0926, 0.5765, 0.1929, 0.8219, 0.0533, 0.2595],
         [0.0118, 0.6234, 0.9052, 0.9021, 0.5443, 0.9826],
         [0.3294, 0.8859, 0.2642, 0.4821, 0.2936, 0.7825],
         [0.0916, 0.6130, 0.3258, 0.7555, 0.2906, 0.3834],
         [0.9492, 0.9900, 0.3934, 0.6345, 0.3282, 0.3203],
         [0.4193, 0.3491, 0.0120, 0.3940, 0.0949, 0.2586]],

        [[0.3945, 0.6685, 0.7638, 0.4580, 0.4213, 0.7744],
         [0.4684, 0.1031, 0.7223, 0.3876, 0.9374, 0.6348],
         [0.5073, 0.0811, 0.9718, 0.3441, 0.9789, 0.5309],
         [0.0464, 0.1444, 0.0131, 0.1354, 0.1363, 0.4884],
         [0.5902, 0.5752, 0.5572, 0.4773, 0.1481, 0.4861],
         [0.4762, 0.7723, 0.5584, 0.3330, 0.9923, 0.3006],
         [0.6967, 0.6267, 0.0114, 0.7994, 0.4867, 0.3571],
         [0.3426, 0.3432, 0.9019, 0.3500, 0.2407, 0.7816]]])
i tensor([[[0.3526, 0.6836, 0.5706, 0.5813, 0.2053, 0.8642],
         [0.4879, 0.3136, 0.8652, 0.7691, 0.5845, 0.8990],
         [0.4399, 0.8685, 0.5658, 0.3264, 0.8709, 0.9613],
         [0.0199, 0.5939, 0.9009, 0.9864, 0.6581, 0.1774],
         [0.4650, 0.2194, 0.5409, 0.1912, 0.4927, 0.9850],
         [0.0033, 0.4131, 0.5793, 0.2638, 0.8113, 0.3437],
         [0.8758, 0.4926, 0.2402, 0.8811, 0.6346, 0.3666],
         [0.4559, 0.6895, 0.2985, 0.8002, 0.9424, 0.0689]],

        [[0.1367, 0.3218, 0.4103, 0.9068, 0.1289, 0.4846],
         [0.0283, 0.4955, 0.3459, 0.5642, 0.7386, 0.5170],
         [0.4937, 0.4622, 0.2706, 0.4797, 0.9714, 0.5258],
         [0.5382, 0.6232, 0.8357, 0.2047, 0.2334, 0.5084],
         [0.0239, 0.3009, 0.2565, 0.4092, 0.3953, 0.5668],
         [0.3464, 0.1299, 0.7938, 0.9157, 0.7993, 0.3227],
         [0.7788, 0.2640, 0.0140, 0.1054, 0.1661, 0.2205],
         [0.4375, 0.0124, 0.2400, 0.9916, 0.7229, 0.1512]]])
'''