RngReadAndSkip

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

Nâng cao bộ đếm của RNG dựa trên bộ đếm.

Trạng thái của RNG sau `rng_read_and_skip(n)` sẽ giống như trạng thái sau `uniform([n])` (hoặc bất kỳ phân phối nào khác). Mức tăng thực tế được thêm vào bộ đếm là một lựa chọn triển khai không xác định.

Trong trường hợp thuật toán đầu vào là RNG_ALG_AUTO_SELECT, bộ đếm ở trạng thái cần có kích thước int64[2], kích thước bộ đếm tối đa hiện tại trong số các thuật toán. Trong trường hợp này, op này sẽ quản lý bộ đếm như thể nó là số nguyên 128 bit có bố cục [low_64bits, Higher_64bits]. Nếu một thuật toán cần ít hơn 128 bit cho bộ đếm thì nó nên sử dụng phần bên trái của int64[2]. Bằng cách này, int64[2] tương thích với tất cả các thuật toán RNG hiện tại (Philox, ThreeFry và xla::RandomAlgorithm::RNG_DEFAULT). Do đó, các hoạt động RNG xuôi dòng có thể sử dụng bộ đếm này với bất kỳ thuật toán RNG nào.

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

Đầu ra <Dài>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
RngReadAndSkip tĩnh
tạo ( Phạm vi phạm vi, tài nguyên Toán hạng <?>, Toán hạng <Số nguyên> alg, Toán hạng <?> delta)
Phương thức gốc để tạo một lớp bao bọc thao tác RngReadAndSkip mới.
Đầu ra <Dài>
giá trị ()
Giá trị cũ của biến tài nguyên, trước khi tăng.

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

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

Đầu ra công khai <Dài> 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.

công khai tĩnh RngReadAndSkip tạo ( Phạm vi phạm vi , tài nguyên Toán hạng <?>, Toán hạng <Số nguyên> alg, Toán hạng <?> delta)

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

Thông số
phạm vi phạm vi hiện tại
nguồn Phần xử lý của biến tài nguyên lưu trữ trạng thái của RNG. Trạng thái bao gồm bộ đếm theo sau là khóa.
tảo Thuật toán RNG.
đồng bằng Số lượng thăng tiến.
Trả lại
  • một phiên bản mới của RngReadAndSkip

giá trị đầu ra công khai <Dài> ()

Giá trị cũ của biến tài nguyên, trước khi tăng. Vì kích thước trạng thái phụ thuộc vào thuật toán nên đầu ra này sẽ được đệm bên phải bằng các số 0 để đạt được hình dạng int64[3] (kích thước trạng thái tối đa hiện tại trong số các thuật toán).