ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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