Tính toán kích thước trọng số có thể được sử dụng bởi mô hình Cudnn RNN.
Trả về kích thước thông số mà mô hình Cudnn RNN có thể sử dụng. Việc phân bổ và khởi tạo trọng số tiếp theo nên sử dụng kích thước này.
num_layers: Chỉ định số lớp trong mô hình RNN. num_units: Chỉ định kích thước của trạng thái ẩn. input_size: Chỉ định kích thước của trạng thái đầu vào. rnn_mode: Cho biết loại mô hình RNN. input_mode: Cho biết liệu có phép chiếu tuyến tính giữa đầu vào và Tính toán thực tế trước lớp đầu tiên hay không. 'skip_input' chỉ được phép khi input_size == num_units; 'auto_select' ngụ ý 'skip_input' khi input_size == num_units; mặt khác, nó ngụ ý 'tuyến tính_input'. hướng: Cho biết liệu mô hình hai chiều có được sử dụng hay không. dir = (hướng == hai chiều)? Tỷ lệ bỏ học 2 : 1: xác suất bỏ học. Khi được đặt thành 0., tính năng bỏ học bị vô hiệu hóa. hạt giống: phần đầu tiên của hạt giống để khởi tạo việc bỏ học. Seed2: phần thứ 2 của hạt giống để khởi tạo việc bỏ học. params_size: Kích thước của bộ đệm params cần được phân bổ và khởi tạo cho mô hình RNN này. Lưu ý rằng bộ đệm thông số này có thể không tương thích trên các GPU. Vui lòng sử dụng CudnnRNNParamsWeights và CudnnRNNParamsBiases để lưu và khôi phục chúng theo cách tương thích trên các lần chạy khác nhau.
Các lớp lồng nhau
lớp học | CudnnRnnParamsSize.Options | Thuộc tính tùy chọn cho CudnnRnnParamsSize |
Hằng số
Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công cộng
Đầu ra <U> | asOutput () Trả về tay cầm tượng trưng của tensor. |
tĩnh <U mở rộng TNumber , T mở rộng TNumber > CudnnRnnParamsSize <U> | |
tĩnh CudnnRnnParamsSize.Options | hướng (Hướng chuỗi) |
tĩnh CudnnRnnParamsSize.Options | bỏ học (Bỏ thả nổi) |
tĩnh CudnnRnnParamsSize.Options | inputMode (Chế độ nhập chuỗi) |
tĩnh CudnnRnnParamsSize.Options | numProj (numProj dài) |
Đầu ra <U> | |
tĩnh CudnnRnnParamsSize.Options | rnnMode (Chuỗi rnnMode) |
tĩnh CudnnRnnParamsSize.Options | hạt giống (Hạt dài) |
tĩnh CudnnRnnParamsSize.Options | hạt giống2 (Hạt dài2) |
Phương pháp kế thừa
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Phương pháp công cộng
Đầu ra công khai <U> asOutput ()
Trả về tay cầm tượng trưng của tensor.
Đầ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.
public static CudnnRnnParamsSize <U> tạo ( Phạm vi phạm vi, Toán hạng < TInt32 > numLayers, Toán hạng < TInt32 > numUnits, Toán hạng < TInt32 > inputSize, Class<T> T, Class<U> S, Tùy chọn... tùy chọn)
Phương thức gốc để tạo một lớp bao bọc một thao tác CudnnRnnParamsSize mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của CudnnRnnParamsSize