一、方法詳解

含義：將一個(gè)張量分為幾個(gè)chunks

torch.split(tensor, split_size_or_sections, dim=0)

• tensor：要分的張量

• split_size_or_sections：

  • 如果該項(xiàng)參數(shù)的值為一個(gè)int類(lèi)型的value值，那么該方法會(huì)將tensor劃分為同等數(shù)量的張量；如果tensor的size沿著給定的不能整除split_size，那么最后一個(gè)chunk相較于其它c(diǎn)hunk小；
  • 如果是一個(gè)list列表，該方法會(huì)將tensor劃分為len(split_size_or_sections)的張量。

• dim：劃分張量所依據(jù)的維度

• return：返回的是一個(gè)tuple

這樣純靠文字解釋，是很抽象的，我們直接通過(guò)以下的案例來(lái)對(duì)這個(gè)方法進(jìn)行徹底的掌握。

二、案例

• 案例1

import torch# 創(chuàng)建一個(gè)張量x = torch.arange(10).reshape(5, 2)print(x)

y = torch.split(x, 2) print(y, type(y))

• 案例2

x = torch.arange(10).reshape(5, 2) print(x) y = torch.split(x, [2, 3]) print(y)

• 案例3

x = torch.arange(10).reshape(5, 2) print(x) y = torch.split(x, [1, 4]) print(y)

相信大家通過(guò)以上這3個(gè)案例，應(yīng)該可以掌握Torch.split()這個(gè)方法了，最后希望對(duì)大家有所幫助！

總結(jié)

以上是生活随笔為你收集整理的PyTorch基础（八）----- torch.split()方法的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。