TBfloat16

공개 인터페이스 TBfloat16

Brain 16비트 부동 소수점 텐서 유형.

이 유형은 32비트 부동 소수점의 가수를 자르고 더 빠른 변환을 위해 모든 지수 비트를 보존한다는 점에서 TFloat16 과 다릅니다. 반면 후자는 더 높은 정밀도를 위해 지수를 축소하고 더 ​​긴 가수를 갖습니다.

Java에는 16비트에 맞는 부동 소수점 유형이 없으므로 JVM에서 이 유형의 텐서에 쓰거나 읽는 각 32비트 값에 대해 변환(정밀도 손실 가능성 있음)이 필요합니다. 따라서 텐서에서 많은 I/O 작업이 예상되는 경우 가능할 때마다 TFloat32 또는 TFloat64 데이터 유형을 사용하여 작업하면 성능이 향상됩니다.

일부 CPU는 기본적으로 bfloat16 형식을 지원하므로 GPU를 사용하지 않을 때 TFloat16 에 비해 계산 속도가 더 빨라질 수 있습니다.

공개 방법

추상 정적 TBfloat16
scalarOf (부동 소수점 값)
단일 부동 소수점 값을 저장하기 위해 새 텐서를 할당합니다.
추상 정적 TBfloat16
tensorOf ( 모양 모양 , 소비자< TBfloat16 > dataInit)
주어진 형태의 새로운 텐서를 할당하고 해당 데이터를 초기화합니다.
추상 정적 TBfloat16
tensorOf ( 모양 모양)
주어진 모양의 새로운 텐서를 할당합니다.
추상 정적 TBfloat16
tensorOf ( 모양 모양, FloatDataBuffer 데이터)
제공된 데이터로 초기화된 지정된 모양의 새 텐서를 할당합니다.
추상 정적 TBfloat16
tensorOf ( NdArray <Float> src)
주어진 부동 소수점 배열의 복사본인 새 텐서를 할당합니다.
추상 정적 TBfloat16
벡터오브 (부동... 값)
부동 소수점 벡터를 저장하기 위해 새 텐서를 할당합니다.

상속된 메서드

공개 방법

공개 정적 추상 TBfloat16 scalarOf (부동 소수점 값)

단일 부동 소수점 값을 저장하기 위해 새 텐서를 할당합니다.

매개변수
새로운 텐서에 저장할 float
보고
  • 새로운 텐서

공개 정적 추상 TBfloat16 tensorOf ( 모양 모양, 소비자< TBfloat16 > dataInit)

주어진 형태의 새로운 텐서를 할당하고 해당 데이터를 초기화합니다.

매개변수
모양 할당할 텐서의 모양
데이터 초기화 텐서 데이터 초기화
보고
  • 새로운 텐서
던지기
TensorFlow예외 텐서를 할당하거나 초기화할 수 없는 경우

공개 정적 추상 TBfloat16 tensorOf ( 모양 모양)

주어진 모양의 새로운 텐서를 할당합니다.

매개변수
모양 할당할 텐서의 모양
보고
  • 새로운 텐서

공개 정적 추상 TBfloat16 tensorOf ( 모양 모양, FloatDataBuffer 데이터)

제공된 데이터로 초기화된 지정된 모양의 새 텐서를 할당합니다.

매개변수
모양 할당할 텐서의 모양
데이터 텐서를 초기화하기 위한 부동 소수점 버퍼
보고
  • 새로운 텐서

공개 정적 추상 TBfloat16 tensorOf ( NdArray <Float> src)

주어진 부동 소수점 배열의 복사본인 새 텐서를 할당합니다.

텐서는 소스 배열과 동일한 모양을 가지며 해당 데이터가 복사됩니다.

매개변수
소스 새 텐서에 모양과 데이터를 제공하는 소스 배열
보고
  • 새로운 텐서

공개 정적 추상 TBfloat16 벡터Of (부동... 값)

부동 소수점 벡터를 저장하기 위해 새 텐서를 할당합니다.

매개변수
가치 새 텐서에 저장하기 위한 부동 소수점
보고
  • 새로운 텐서