このページは Cloud Translation API によって翻訳されました。
Switch to English

tf.one_hot

TensorFlow 1つのバージョン GitHubの上のソースを表示

ワンホットテンソルを返します。

ノートPCで使用されます

ガイドで使用チュートリアルで使用されます

参照してくださいtf.filltf.eye

指数で表される位置indices値を取るon_value全ての他の位置の値がとりながら、 off_value

on_valueoff_valueデータ型が一致している必要があります。場合dtypeも提供することで指定されているように、彼らは同じデータ型でなければなりませんdtype

場合on_value提供されていない、それが値にデフォルト設定されます1型とdtype

場合off_value提供されていない、それは値がデフォルトになります0型でdtype

入力された場合にはindicesランクであるN 、出力は、ランクがありますN+1 。新しい軸は、寸法で作成されたaxis (:新しい軸は、最後に追加されるデフォルト)。

場合indicesスカラーである出力形状は、長さのベクトルとなりdepth

場合indices 、長さのベクトルであるfeatures 、出力形状は次のようになります。

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

場合indices形状のマトリックス(バッチ)である[batch, features] 、出力形状は次のようになります。

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

場合はindices RaggedTensorで、「軸」引数は正でかつ非ギザギザの軸を参照する必要があります。出力はRaggedTensorの値に「one_hot」を適用し、その結果から新しいRaggedTensorを作成することと同じになります。

場合dtype提供されていない、それはのデータ型と仮定しようとしますon_valueoff_valueのどれ場合一方または両方が渡されている場合は、。 on_valueoff_value 、またはdtype提供されていない、 dtype値がデフォルトになりますtf.float32

例えば:

 indices = [0, 1, 2]
depth = 3
tf.one_hot(indices, depth)  # output: [3 x 3]
# [[1., 0., 0.],
#  [0., 1., 0.],
#  [0., 0., 1.]]

indices = [0, 2, -1, 1]
depth = 3
tf.one_hot(indices, depth,
           on_value=5.0, off_value=0.0,
           axis=-1)  # output: [4 x 3]
# [[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)

indices = [[0, 2], [1, -1]]
depth = 3
tf.one_hot(indices, depth,
           on_value=1.0, off_value=0.0,
           axis=-1)  # output: [2 x 2 x 3]
# [[[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)

indices = tf.ragged.constant([[0, 1], [2]])
depth = 3
tf.one_hot(indices, depth)  # output: [2 x None x 3]
# [[[1., 0., 0.],
#   [0., 1., 0.]],
#  [[0., 0., 1.]]]
 

indices Tensorのインデックスの。
depth 1つのホット次元の深さを定義するスカラー。
on_value ときに出力を埋めるために値を定義するスカラーindices[j] = i 。 (デフォルト:1)
off_value とき出力を埋めるために値を定義するスカラーindices[j] != i 。 (デフォルト:0)
axis 軸は(デフォルト:-1、新しい最も内側の軸)を記入します。
dtype 出力テンソルのデータ型。
name 操作の名前(オプション)。

output ワンホットテンソル。

TypeError いずれかのDTYPE場合on_valueまたはoff_value一致しないdtype
TypeError DTYPEもしon_valueoff_value互いに一致しません