Cập nhật các mục nhập có liên quan trong '*var' và '*accum' theo sơ đồ động lượng.
Đặt use_nesterov = True nếu bạn muốn sử dụng đà Nesterov.
Đó là đối với các hàng chúng tôi có grad, chúng tôi cập nhật var và accum như sau:
$$accum = accum * momentum + grad$$$$var -= lr * accum$$
Các lớp lồng nhau
lớp học | SparseApplyMomentum.Options | Thuộc tính tùy chọn cho SparseApplyMomentum |
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 khai
Đầu ra <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
tĩnh <T mở rộng TType > SparseApplyMomentum <T> | tạo ( Phạm vi phạm vi , Toán hạng <T> var, Toán hạng <T> accum, Toán hạng <T> lr, Toán hạng <T> grad, Toán hạng <? mở rộng TNumber > chỉ số, Động lượng Toán hạng <T>, Tùy chọn... tùy chọn) Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác SparseApplyMomentum mới. |
Đầu ra <T> | ngoài () Tương tự như "var". |
tĩnh SparseApplyMomentum.Options | useLocking (Boolean useLocking) |
tĩnh SparseApplyMomentum.Options | useNesterov (Sử dụng BooleanNesterov) |
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 khai
Đầu ra công khai <T> 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 SparseApplyMomentum <T> create ( Phạm vi phạm vi, Toán hạng <T> var, Toán hạng <T> accum, Toán hạng <T> lr, Toán hạng <T> grad, Toán hạng <? mở rộng TNumber > chỉ số, Động lượng toán hạng <T>, Tùy chọn ... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác SparseApplyMomentum mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
var | Phải từ một Biến(). |
tích lũy | Phải từ một Biến(). |
lr | Tỷ lệ học tập. Phải là một vô hướng. |
tốt nghiệp | Độ dốc. |
chỉ số | Một vectơ chỉ số theo chiều thứ nhất của var và accum. |
đà | Động lực. Phải là một vô hướng. |
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 SparseApplyMomentum
public static SparseApplyMomentum.Options useLocking (Boolean useLocking)
Thông số
sử dụngKhóa | Nếu `True`, việc cập nhật các tensor var và accum sẽ được bảo vệ bằng khóa; mặt khác, hành vi không được xác định, nhưng có thể ít gây tranh cãi hơn. |
---|
public static SparseApplyMomentum.Options useNesterov (Boolean useNesterov)
Thông số
sử dụngNesterov | Nếu `True`, tenxơ được truyền để tính grad sẽ là var - lr * đà * tích lũy, vì vậy cuối cùng, var bạn nhận được thực sự là var - lr * xung lượng * tích lũy. |
---|