tensorflow::ops::OneHot::Attrs

#include <array_ops.h>

Optional attribute setters for OneHot.

Summary

Public attributes

axis_ = -1
int64

Public functions

Axis(int64 x)
TF_MUST_USE_RESULT Attrs
The axis to fill (default: -1, a new inner-most axis).

Public attributes

axis_

int64 tensorflow::ops::OneHot::Attrs::axis_ = -1

Public functions

Axis

TF_MUST_USE_RESULT Attrs tensorflow::ops::OneHot::Attrs::Axis(
  int64 x
)

The axis to fill (default: -1, a new inner-most axis).

Defaults to -1