Trả lời cho sự kiện TensorFlow Everywhere tại địa phương của bạn ngay hôm nay!
Trang này được dịch bởi Cloud Translation API.
Switch to English

dòng chảy :: ops :: OneHot

#include <array_ops.h>

Trả về một tensor một nóng.

Tóm lược

Các vị trí được đại diện bởi các chỉ số trong indices nhận giá trị on_value , trong khi tất cả các vị trí khác lấy giá trị off_value .

Nếu các indices đầu vào là thứ hạng N , đầu ra sẽ có thứ hạng N+1 , Trục mới được tạo ở axis nguyên (mặc định: trục mới được nối vào cuối).

Nếu indices là một đại lượng vô hướng thì hình dạng đầu ra sẽ là một vectơ có độ depth chiều dài.

Nếu indices là một vector có độ dài features , hình dạng đầu ra sẽ là:

  features x depth if axis == -1
  depth x features if axis == 0

Nếu indices là một ma trận (lô) có hình dạng [batch, features] , hình dạng đầu ra sẽ là:

  batch x features x depth if axis == -1
  batch x depth x features if axis == 1
  depth x batch x features if axis == 0

Ví dụ

Giả sử rằng

  indices = [0, 2, -1, 1]
  depth = 3
  on_value = 5.0
  off_value = 0.0
  axis = -1

Sau đó, đầu ra là [4 x 3] :

output =
  [5.0 0.0 0.0]  // one_hot(0)
  [0.0 0.0 5.0]  // one_hot(2)
  [0.0 0.0 0.0]  // one_hot(-1)
  [0.0 5.0 0.0]  // one_hot(1)

Giả sử rằng

  indices = [0, 2, -1, 1]
  depth = 3
  on_value = 0.0
  off_value = 3.0
  axis = 0

Sau đó, đầu ra là [3 x 4] :

output =
  [0.0 3.0 3.0 3.0]
  [3.0 3.0 3.0 0.0]
  [3.0 3.0 3.0 3.0]
  [3.0 0.0 3.0 3.0]
//  ^                one_hot(0)
//      ^            one_hot(2)
//          ^        one_hot(-1)
//              ^    one_hot(1)

Giả sử rằng

  indices = [[0, 2], [1, -1]]
  depth = 3
  on_value = 1.0
  off_value = 0.0
  axis = -1

Sau đó, đầu ra là [2 x 2 x 3] :

output =
  [
    [1.0, 0.0, 0.0]  // one_hot(0)
    [0.0, 0.0, 1.0]  // one_hot(2)
  ][
    [0.0, 1.0, 0.0]  // one_hot(1)
    [0.0, 0.0, 0.0]  // one_hot(-1)
  ]

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • chỉ số: Một hàng chục chỉ số.
  • độ sâu: Một đại lượng vô hướng xác định độ sâu của một chiều nóng.
  • on_value: Một đại lượng vô hướng xác định giá trị để điền vào đầu ra khi indices[j] = i .
  • off_value: Một đại lượng vô hướng xác định giá trị để điền vào đầu ra khi indices[j] != i .

Các thuộc tính tùy chọn (xem Phần Attrs ):

  • axis: Trục cần điền (mặc định: -1, trục trong cùng mới).

Lợi nhuận:

  • Output : Bộ căng một nóng.

Người xây dựng và Người phá hủy

OneHot (const :: tensorflow::Scope & scope, :: tensorflow::Input indices, :: tensorflow::Input depth, :: tensorflow::Input on_value, :: tensorflow::Input off_value)
OneHot (const :: tensorflow::Scope & scope, :: tensorflow::Input indices, :: tensorflow::Input depth, :: tensorflow::Input on_value, :: tensorflow::Input off_value, const OneHot::Attrs & attrs)

Thuộc tính công cộng

operation
output

Chức năng công cộng

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Chức năng tĩnh công khai

Axis (int64 x)

Cấu trúc

tensorflow :: ops :: OneHot :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho OneHot .

Thuộc tính công cộng

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

Chức năng công cộng

OneHot

 OneHot(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input indices,
  ::tensorflow::Input depth,
  ::tensorflow::Input on_value,
  ::tensorflow::Input off_value
)

OneHot

 OneHot(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input indices,
  ::tensorflow::Input depth,
  ::tensorflow::Input on_value,
  ::tensorflow::Input off_value,
  const OneHot::Attrs & attrs
)

nút

::tensorflow::Node * node() const 

toán tử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const 

toán tử :: tensorflow :: Đầu ra

 operator::tensorflow::Output() const 

Chức năng tĩnh công khai

Trục

Attrs Axis(
  int64 x
)