dòng chảy căng:: ôi:: TrongTopKV2

Cho biết liệu các mục tiêu có nằm trong dự đoán K hàng đầu hay không.

Bản tóm tắt

Điều này tạo ra một mảng bool batch_size , một mục nhập out[i] là true nếu dự đoán cho lớp mục tiêu nằm trong số k dự đoán hàng đầu trong số tất cả các dự đoán, ví dụ i . Lưu ý rằng hoạt động của InTopK khác với hoạt động TopK ở cách xử lý các mối quan hệ; nếu nhiều lớp có cùng giá trị dự đoán và nằm trên ranh giới top- k thì tất cả các lớp đó được coi là nằm trong top k .

Chính thức hơn, hãy

\(predictions_i\) là dự đoán cho tất cả các lớp, ví dụ i , \(targets_i\) là lớp mục tiêu chẳng hạn i , \(out_i\) là đầu ra chẳng hạn i ,

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

Lập luận:

• phạm vi: Một đối tượng Phạm vi
• dự đoán: Một tensor batch_size x classes .
• mục tiêu: Một vectơ batch_size của id lớp.
• k: Số phần tử hàng đầu cần xem xét để có độ chính xác tính toán.

Trả về:

• Output : Độ chính xác được tính toán ở k dưới dạng bool Tensor .

Thuộc tính công khai

Operation operation

::tensorflow::Output precision

 InTopKV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  ::tensorflow::Input k
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const