torch.reshape(): Reorganizarea Robustă a Formei Tensorului în PyTorch & Deep Learning Practic

Funcția torch.reshape() (și metoda asociată tensor.reshape()) returnează un tensor cu aceeași cantitate totală de date, dar reorganizat după noua formă specificată. Dacă tensorul este contiguu va întoarce o vedere (fără copiere), iar dacă nu este contiguu va clona datele automat.

Exemplu de utilizare

import torch

t = torch.randn(4, 3, 2)
# Reorganizare în lot bidimensional
t_2d = t.reshape(4, -1)
print(t_2d.shape)  # torch.Size([4, 6])

Este recomandat în detrimentul lui .view() atunci când codul manipulează tensori rezultați din permutări sau transpuneri.