DeviceIndex

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

Trả về chỉ mục của thiết bị mà op chạy.

Đưa ra danh sách tên thiết bị, thao tác này trả về chỉ mục của thiết bị mà hoạt động này chạy. Độ dài của danh sách được trả về trong hai trường hợp: (1) Thiết bị không tồn tại trong danh sách thiết bị đã cho. (2) Nó nằm trong phần biên dịch XLA.

Phương pháp công cộng

Đầu ra <Số nguyên>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
Chỉ số thiết bị tĩnh
tạo ( Phạm vi phạm vi, Danh sách<String> deviceNames)
Phương thức gốc để tạo một lớp bao bọc một thao tác DeviceIndex mới.
Đầu ra <Số nguyên>

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

Phương pháp công cộng

Đầu ra công khai <Số nguyên> asOutput ()

Trả về phần điều khiển tượng trưng của một tenxơ.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

tạo chỉ số thiết bị tĩnh công khai ( Phạm vi phạm vi, Danh sách<String> tên thiết bị)

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

Thông số
phạm vi phạm vi hiện tại
Trả lại
  • một phiên bản mới của DeviceIndex

chỉ số đầu ra công khai <Integer> ()