Fonction torch.split_with_sizes de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.split_with_sizesC'est une fonction de PyTorch utilisée pour diviser un tenseur selon les tailles spécifiées.

Définition de fonction

torch.split_with_sizes(input, split_sizes, dim=0)

Exemple d'utilisation

Exemple

import torch

x = torch.arange(10)
print("Tenseur d'origine :")
print(x)

# Diviser selon les tailles [2, 3, 5]
result = torch.split_with_sizes(x, [2, 3, 5])
print("Résultats de la division :")
for i, t in enumerate(result):
    print(f" Segment {i} : {t}")

# Diviser un tenseur 2D par lignes
y = torch.arange(12).reshape(4, 3)
print("nTenseur 2D d'origine :")
print(y)

result = torch.split_with_sizes(y, [1, 2, 1], dim=0)
print("Découpage par lignes [1, 2, 1]:")
for i, t in enumerate(result):
    print(f" Bloc {i}:n{t}")

La sortie est :

原始张量:
tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
分割结果:
  块 0: tensor([0, 1])
  Chunk 1: tensor([2, 3, 4])
  Chunk 2: tensor([5, 6, 7, 8, 9])

原始二维张量:
tensor([[ 0,  1,  2],
        [ 3,  4,  5],
        [ 6,  7,  8],
        [ 9, 10, 11]])
按行分割 [1, 2, 1]:
  块 0:
tensor([[0, 1, 2]])
  Chunk 1:
tensor([[3, 4, 5],
        [6, 7, 8]])
  Chunk 2:
tensor([[ 9, 10, 11]])

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions