DeviceIndex

공개 최종 클래스 DeviceIndex

작업이 실행되는 장치의 인덱스를 반환합니다.

장치 이름 목록이 주어지면 이 작업은 이 작업이 실행하는 장치의 인덱스를 반환합니다. 목록의 길이는 두 가지 경우에 반환됩니다. (1) 해당 장치 목록에 장치가 존재하지 않습니다. (2) XLA 컴파일에 있습니다.

공개 방법

출력 <정수>
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 DeviceIndex
생성 ( 범위 범위, List<String> deviceNames)
새 DeviceIndex 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <정수>
색인 ()

상속된 메서드

공개 방법

공개 출력 <Integer> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

공개 정적 DeviceIndex 생성 ( 범위 범위, List<String> deviceNames)

새 DeviceIndex 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
보고
  • DeviceIndex의 새 인스턴스

공개 출력 <Integer> 인덱스 ()