TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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