lớp cuối cùng công khai ArgMax
Trả về chỉ mục có giá trị lớn nhất trên các kích thước của tensor.
Lưu ý rằng trong trường hợp ràng buộc, danh tính của giá trị trả về không được đảm bảo.
Cách sử dụng:
import tensorflow as tf
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
b = tf.math.argmax(input = a)
c = tf.keras.backend.eval(b)
# c = 4
# here a[4] = 166.32 which is the largest element of a across axis 0
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 <V> | asOutput () Trả về tay cầm tượng trưng của tensor. |
| tĩnh <V mở rộng TNumber > ArgMax <V> | |
| ArgMax tĩnh < TInt64 > | |
| Đầu ra <V> | đầu ra () |
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
Giá trị không đổi: "ArgMax"
Phương pháp công khai
Đầu ra công khai <V> 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 ArgMax <V> tạo ( Phạm vi phạm vi, Toán hạng <? mở rộng TType > đầu vào, Toán hạng <? mở rộng TNumber > kích thước, Lớp<V> loại đầu ra)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác ArgMax mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| kích thước | int32 hoặc int64, phải nằm trong phạm vi `[-rank(input), Rank(input))`. Mô tả kích thước nào của Tensor đầu vào cần giảm. Đối với vectơ, sử dụng thứ nguyên = 0. |
Trả lại
- một phiên bản mới của ArgMax
public static ArgMax < TInt64 > tạo ( Phạm vi phạm vi, Toán hạng <? mở rộng TType > đầu vào, Toán hạng <? mở rộng TNumber > kích thước)
Phương thức gốc để tạo một lớp bao bọc một thao tác ArgMax mới bằng cách sử dụng các loại đầu ra mặc định.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| kích thước | int32 hoặc int64, phải nằm trong phạm vi `[-rank(input), Rank(input))`. Mô tả kích thước nào của Tensor đầu vào cần giảm. Đối với vectơ, sử dụng thứ nguyên = 0. |
Trả lại
- một phiên bản mới của ArgMax