Xem bài phát biểu, phiên sản phẩm, hội thảo và hơn thế nữa từ Google I / O Xem danh sách phát

dòng chảy :: hoạt động :: OneHot

#include <array_ops.h>

Trả về 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, 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à 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

Các 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 khai

operation
output

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

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

Các 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 khai

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 

Các chức năng tĩnh công khai

Trục

Attrs Axis(
  int64 x
)