1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Nhập môn Deep Learning với PyTorch

Connected

Bài tập

Sử dụng TensorDataset

Cấu trúc dữ liệu của bạn thành một dataset là một trong những bước đầu tiên khi huấn luyện mạng nơ-ron PyTorch. TensorDataset giúp đơn giản hóa việc này bằng cách chuyển đổi mảng NumPy sang định dạng mà PyTorch có thể dùng.

Trong bài tập này, bạn sẽ tạo một TensorDataset từ bộ dữ liệu animals đã được nạp sẵn và kiểm tra cấu trúc của nó.

Hướng dẫn

100 XP
  • Chuyển X và y thành tensor và tạo một TensorDataset.
  • Truy cập và in ra mẫu đầu tiên.