Ajustarea Dimensiunilor Unitare: squeeze și expand_dims în NumPy

Adăugarea sau eliminarea axelor cu lungimea 1 pentru alinierea dimensiunilor în modele de Deep Learning.

Sintaxă
np.expand_dims(a, axis) / np.squeeze(a, axis=None)

La pregătirea datelor pentru rețele neuronale (PyTorch, TensorFlow), tensorii necesită adesea dimensiuni explicite de lot (batch size):

  • np.expand_dims(a, axis): Inserează o nouă axă cu lungimea 1 (transformă un vector 1D de formă (100,) într-o matrice 2D (1, 100)).
  • np.squeeze(a): Elimină toate axele redundante care au dimensiunea 1.
Exemplu: Pregătirea unei imagini pentru inferență
import numpy as np

imagine = np.zeros((28, 28))  # Imagine 28x28
print("Formă inițială:", imagine.shape)

# Adăugăm dimensiunea de batch pe poziția 0:
batch = np.expand_dims(imagine, axis=0)
print("Formă cu batch (1, 28, 28):", batch.shape)

# Revenire la forma fără axă unitară:
inapoi = np.squeeze(batch)
print("Formă după squeeze:", inapoi.shape)

Comunitate & Schimb de cunoștințe

Întrebări & Exemple de Cod: Ajustarea Dimensiunilor Unitare: squeeze și expand_dims

Ai o nelămurire despre utilizarea comenzii Ajustarea Dimensiunilor Unitare: squeeze și expand_dims sau ai descoperit un truc / fragment de cod util? Distribuie-l cu ceilalți programatori!

💬

Fii primul care pune o întrebare sau distribuie un fragment de cod / prompt!

Nu au fost adăugate încă întrebări sau exemple de cod pentru această pagină. Dacă ai întâlnit o problemă sau ai un snippet util, adaugă-l mai jos pentru a-i ajuta pe ceilalți.

Comunitate Adaugă o întrebare sau distribuie un cod

Moderare & Siguranță:

Pentru a menține calitatea și securitatea informațiilor, fiecare întrebare sau fragment de cod/prompt este analizat (manual sau prin moderare asistată de AI) înainte de publicare.