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 )