Program Listing for File adaptive.h

↰ Return to documentation for file (torch/csrc/api/include/torch/nn/options/adaptive.h)

#pragma once

#include <torch/arg.h>
#include <torch/csrc/Export.h>
#include <torch/types.h>

namespace torch {
namespace nn {

struct TORCH_API AdaptiveLogSoftmaxWithLossOptions {
  /* implicit */ AdaptiveLogSoftmaxWithLossOptions(
      int64_t in_features,
      int64_t n_classes,
      std::vector<int64_t> cutoffs);

  TORCH_ARG(int64_t, in_features);

  TORCH_ARG(int64_t, n_classes);

  TORCH_ARG(std::vector<int64_t>, cutoffs);

  TORCH_ARG(double, div_value) = 4.;

  TORCH_ARG(bool, head_bias) = false;
};

} // namespace nn
} // namespace torch