XlaSparseCoreAdam

공개 최종 클래스 XlaSparseCoreAdam

공개 방법

정적 XlaSparseCoreAdam
create ( Scope 범위, Operand <Float> embeddingTable, Operand <Integer> 인덱스, Operand <Float> 그래디언트, Operand <Float> learningRate, Operand <Float> 모멘텀, Operand <Float> 속도, Operand <Float> beta1, Operand <Float > beta2, 피연산자 <Float> 엡실론, Long featureWidth, Boolean useSumInsideSqrt)
새로운 XlaSparseCoreAdam 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <플로트>
출력 <플로트>
출력 <플로트>

상속된 메서드

공개 방법

public static XlaSparseCoreAdam create ( Scope 범위, Operand <Float> embeddingTable, Operand <Integer> 인덱스, Operand <Float> 그래디언트, Operand <Float> learningRate, Operand <Float> 추진력, Operand <Float> 속도, Operand <Float> beta1, 피연산자 <Float> beta2, 피연산자 <Float> 엡실론, Long featureWidth, Boolean useSumInsideSqrt)

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

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

공개 출력 <Float> updateEmbeddingTable ()

공개 출력 <Float> 업데이트된 모멘텀 ()

공개 출력 <Float> updateVelocity ()