Funcția torch.ones() alocă și returnează un tensor cu toate valorile inițializate la 1. Este folosită frecvent la generarea măștilor de atenție implicite sau a tensorilor de multiplicare de bază.
Exemplu practic
import torch
# Crearea unui tensor 2x2x3 de 1 pe GPU dacă este disponibil
device = "cuda" if torch.cuda.is_available() else "cpu"
masca = torch.ones((2, 2, 3), dtype=torch.bool, device=device)
print(masca)Similar, torch.ones_like(input) copiază automat forma și tipul de date ale tensorului de intrare.
Întrebări & Exemple practice:
torch.ones(): Crearea Tensorilor Umpluți cu UnuAi o întrebare despre torch.ones(): Crearea Tensorilor Umpluți cu Unu sau ai descoperit un truc, prompt sau fragment de cod util? Distribuie-l cu ceilalți membri ai comunității!
Fii primul care pune o întrebare sau distribuie un fragment de cod / prompt!
Nu au fost adăugate încă întrebări sau resurse pentru această pagină. Dacă ai testat un prompt sau ai un snippet util, adaugă-l mai jos pentru comunitate.