Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.modules.pooling.AvgPool1d Class Reference
Inheritance diagram for torch.nn.modules.pooling.AvgPool1d:
torch.nn.modules.pooling._AvgPoolNd

Public Member Functions

def __init__ (self, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True)
 
def forward (self, input)
 
- Public Member Functions inherited from torch.nn.modules.pooling._AvgPoolNd
def extra_repr (self)
 

Public Attributes

 kernel_size
 
 stride
 
 padding
 
 ceil_mode
 
 count_include_pad
 

Detailed Description

Definition at line 445 of file pooling.py.


The documentation for this class was generated from the following file: