InfeedEnqueueTuple

lớp cuối cùng công khai InfeedEnqueueTuple

Đưa nhiều giá trị Tensor vào tính toán dưới dạng bộ dữ liệu XLA.

Các lớp lồng nhau

lớp học InfeedEnqueueTuple.Options Thuộc tính tùy chọn cho InfeedEnqueueTuple

Phương pháp công khai

InfeedEnqueueTuple tĩnh
tạo ( Phạm vi phạm vi, Iterable< Toán hạng <?>> đầu vào, Danh sách< Hình dạng > hình dạng, Tùy chọn... tùy chọn)
Phương thức gốc để tạo một lớp bao bọc một thao tác InfeedEnqueueTuple mới.
InfeedEnqueueTuple.Options tĩnh
deviceOrdinal (Thiết bị dàiOrdinal)
InfeedEnqueueTuple.Options tĩnh
bố cục (Bố cục danh sách<Long>)

Phương pháp kế thừa

Phương pháp công khai

public static InfeedEnqueueTuple tạo ( Phạm vi phạm vi, Iterable< Toán hạng <?>> đầu vào, Danh sách< Hình dạng > hình dạng, Tùy chọn... tùy chọn)

Phương thức gốc để tạo một lớp bao bọc một thao tác InfeedEnqueueTuple mới.

Thông số
phạm vi phạm vi hiện tại
đầu vào Danh sách các tensor sẽ được cung cấp bằng cơ chế nạp liệu.
hình dạng Hình dạng của mỗi tenxơ trong `đầu vào`.
tùy chọn mang các giá trị thuộc tính tùy chọn
Trả lại
  • một phiên bản mới của InfeedEnqueueTuple

public static InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)

Thông số
thiết bịthứ tự Thiết bị TPU sẽ sử dụng. Giá trị này phải là -1 khi Op đang chạy trên thiết bị TPU và >= 0 khi Op đang chạy trên thiết bị CPU.

bố cục InfeedEnqueueTuple.Options tĩnh công khai (Bố cục danh sách<Long>)

Thông số
bố cục Một vectơ giữ bố cục được yêu cầu theo trình tự từ nhỏ đến lớn cho tất cả các hình dạng bộ dữ liệu, theo thứ tự các hình dạng xuất hiện trong đầu vào "hình dạng". Các thành phần bố cục cho hình dạng phụ có thể được đặt thành -1, trong trường hợp đó, bố cục tương ứng sẽ được tính toán bằng thao tác nạp liệu.