Bahasa

Referensi Cepat PyTorch

Cheat sheet siap cetak untuk Referensi Cepat PyTorch, disusun untuk pencarian cepat, belajar, dan referensi di meja

Sumber
PyTorch Documentation (pytorch.org)
Lisensi
MIT
Diperbarui
2026-03-27

Contoh

Isi cheat sheet ini

  • Tensor
  • Autograd
  • Neural Network
  • Memuat Data
  • Loop Training
  • Optimizer
  • Fungsi Loss
  • Simpan & Muat
  • GPU
  • Pola Umum

Membuat Tensor

import torch
a = torch.tensor([1, 2, 3])
b = torch.zeros(2, 3)
c = torch.ones(3, 3)
d = torch.randn(2, 4)  # distribusi normal

Konstruktor Tensor

torch.zeros(m, n)Semua nol, shape (m, n)
torch.ones(m, n)Semua satu, shape (m, n)
torch.randn(m, n)Acak distribusi normal standar
torch.arange(start, end, step)Nilai berspasi merata
torch.linspace(start, end, steps)Jumlah titik tetap
torch.eye(n)Matriks identitas
torch.empty(m, n)Memori tidak diinisialisasi

Interop NumPy

t = torch.from_numpy(np_array)
arr = tensor.numpy()  # berbagi memori
t = torch.as_tensor(np_array)

Unduh cheat sheet lengkap dengan semua bagian, siap cetak

Unduh PDF

Unduhan

Pilih varian yang sesuai dengan cara Anda membaca