TBfloat16

giao diện công cộng TBfloat16

Brain loại tensor float 16-bit.

Loại này khác với TFloat16 ở chỗ nó cắt bớt phần định trị của float 32 bit và bảo toàn tất cả các bit số mũ để chuyển đổi nhanh hơn, trong khi loại sau thu nhỏ phần mũ và có phần định trị dài hơn để có độ chính xác cao hơn.

Vì không có loại dấu phẩy động nào phù hợp với 16 bit trong Java, nên cần phải chuyển đổi (có khả năng mất độ chính xác) cho mỗi giá trị 32 bit được ghi hoặc đọc trên một tensor thuộc loại này từ JVM. Do đó, nếu cần thực hiện nhiều thao tác I/O trên một tensor, hiệu suất sẽ được cải thiện bằng cách làm việc với các kiểu dữ liệu TFloat32 hoặc TFloat64 bất cứ khi nào có thể.

Lưu ý rằng một số CPU hỗ trợ định dạng bfloat16 nguyên bản, điều này có thể dẫn đến tính toán nhanh hơn so với TFloat16 khi không sử dụng GPU.

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

trừu tượng tĩnh TBfloat16
vô hướngOf (giá trị float)
Cấp phát một tensor mới để lưu trữ một giá trị float đơn.
trừu tượng tĩnh TBfloat16
tensorOf ( Hình dạng, Người tiêu dùng< TBfloat16 > dataInit)
Cấp phát một tensor mới có hình dạng nhất định và khởi tạo dữ liệu của nó.
trừu tượng tĩnh TBfloat16
tensorOf ( Hình dạng)
Cấp phát một tensor mới có hình dạng đã cho.
trừu tượng tĩnh TBfloat16
tensorOf ( Hình dạng, dữ liệu FloatDataBuffer )
Phân bổ một tensor mới có hình dạng nhất định, được khởi tạo với dữ liệu được cung cấp.
trừu tượng tĩnh TBfloat16
tensorOf ( NdArray <Float> src)
Cấp phát một tensor mới là bản sao của một mảng float nhất định.
trừu tượng tĩnh TBfloat16
vectorOf (giá trị float...)
Cấp phát một tensor mới để lưu trữ một vectơ số float.

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

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

trừu tượng tĩnh công khai TBfloat16 vô hướngOf (giá trị float)

Cấp phát một tensor mới để lưu trữ một giá trị float đơn.

Thông số
giá trị float để lưu trữ trong tensor mới
Trả lại
  • tensor mới

tóm tắt tĩnh công khai TBfloat16 tensorOf ( Hình dạng hình dạng , Người tiêu dùng< TBfloat16 > dataInit)

Cấp phát một tensor mới có hình dạng nhất định và khởi tạo dữ liệu của nó.

Thông số
hình dạng hình dạng của tensor để phân bổ
dữ liệu ban đầu bộ khởi tạo dữ liệu tensor
Trả lại
  • tensor mới
Ném
TensorFlowException nếu tensor không thể được phân bổ hoặc khởi tạo

tóm tắt tĩnh công khai TBfloat16 tensorOf ( Hình dạng hình dạng )

Cấp phát một tenxơ mới có hình dạng đã cho.

Thông số
hình dạng hình dạng của tensor để phân bổ
Trả lại
  • tensor mới

tóm tắt tĩnh công khai TBfloat16 tensorOf (Hình dạng hình dạng , dữ liệu FloatDataBuffer )

Phân bổ một tensor mới có hình dạng nhất định, được khởi tạo với dữ liệu được cung cấp.

Thông số
hình dạng hình dạng của tensor để phân bổ
dữ liệu bộ đệm của số float để khởi tạo tensor với
Trả lại
  • tensor mới

tóm tắt tĩnh công khai TBfloat16 tensorOf ( NdArray <Float> src)

Cấp phát một tensor mới là bản sao của một mảng float nhất định.

Tensor sẽ có hình dạng giống như mảng nguồn và dữ liệu của nó sẽ được sao chép.

Thông số
src mảng nguồn cung cấp hình dạng và dữ liệu cho tensor mới
Trả lại
  • tensor mới

trừu tượng tĩnh công khai TBfloat16 vectorOf (giá trị float...)

Cấp phát một tensor mới để lưu trữ một vectơ số float.

Thông số
giá trị nổi để lưu trữ trong tensor mới
Trả lại
  • tensor mới