docs
0.0-2dfa388-SNAPSHOT
API
Generated with
docs/torch/torch.nn/torch.nn.modules/torch.nn.modules.pooling

torch.nn.modules.pooling

package torch.nn.modules.pooling

Members list

Type members

Classlikes

final class AdaptiveAvgPool2d[D <: BFloat16 | Float32 | Float64](outputSize: Int | Option[Int] | (Option[Int], Option[Int]) | (Int, Int))(implicit evidence$1: Default[D]) extends Module

Applies a 2D adaptive average pooling over an input signal composed of several input planes.

Applies a 2D adaptive average pooling over an input signal composed of several input planes.

The output is of size H x W, for any input size. The number of output features is equal to the number of input planes.

Attributes

Source
AdaptiveAvgPool2d.scala
Supertypes
class Module
class Object
trait Matchable
class Any
final class MaxPool2d[D <: BFloat16 | Float32 | Float64](kernelSize: Int | (Int, Int), stride: Option[Int | (Int, Int)], padding: Int | (Int, Int), dilation: Int | (Int, Int), ceilMode: Boolean)(implicit evidence$1: Default[D]) extends TensorModule[D]

Applies a 2D max pooling over an input signal composed of several input planes.

Applies a 2D max pooling over an input signal composed of several input planes.

Attributes

Source
MaxPool2d.scala
Supertypes
trait TensorModule[D]
trait Tensor[D] => Tensor[D]
class Module
class Object
trait Matchable
class Any
Show all
Generated with