Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::nn::LinearOptions Struct Reference

Options for the Linear module. More...

#include <linear.h>

Public Member Functions

 LinearOptions (int64_t in, int64_t out)
 
 TORCH_ARG (int64_t, in)
 The number of input features (columns of the input matrix).
 
 TORCH_ARG (int64_t, out)
 The number of output features to produce (columns of the output matrix).
 
 TORCH_ARG (bool, with_bias)
 Whether to learn and add a bias after the linear transformation.
 

Detailed Description

Options for the Linear module.

Definition at line 14 of file linear.h.


The documentation for this struct was generated from the following files: