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

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.

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

Operation operation

::tensorflow::Output output

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs Axis(
  int64 x
)