org.bytedeco.pytorch

Class MaxPool2dImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.MaxPool2dImplCloneable

org.bytedeco.pytorch.MaxPool2dImplBase

org.bytedeco.pytorch.MaxPool2dImpl

All Implemented Interfaces:

AutoCloseable

@Namespace(value="torch::nn")
 @Properties(inherit=torch.class)
public class MaxPool2dImpl
extends MaxPool2dImplBase

Applies maxpool over a 2-D input. See https://pytorch.org/docs/master/nn.html#torch.nn.MaxPool2d to learn about the exact behavior of this module. See the documentation for torch::nn::MaxPool2dOptions class to learn what constructor arguments are supported for this module. Example:

  MaxPool2d model(MaxPool2dOptions({3, 2}).stride({2, 2}));
  

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
MaxPool2dImpl(LongPointer kernel_size)
MaxPool2dImpl(MaxPool2dOptions options_)
MaxPool2dImpl(Module pointer) Downcast constructor.
MaxPool2dImpl(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
T_TensorTensor_T	forward_with_indices(Tensor input) Returns the outputs and the indices of the max values.
Tensor	forward(Tensor input)

Methods inherited from class org.bytedeco.pytorch.MaxPool2dImplBase

options, options, pretty_print, reset

Methods inherited from class org.bytedeco.pytorch.MaxPool2dImplCloneable

asModule, asModule, clone, clone

Methods inherited from class org.bytedeco.pytorch.Module

apply, apply, apply, apply, apply, apply, apply, apply, buffers, buffers, children, eval, is_serializable, is_training, load, modules, modules, name, named_buffers, named_buffers, named_children, named_modules, named_modules, named_modules, named_parameters, named_parameters, parameters, parameters, register_buffer, register_buffer, register_module, register_module, register_parameter, register_parameter, register_parameter, register_parameter, save, shiftLeft, to, to, to, train, unregister_module, unregister_module, zero_grad

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

MaxPool2dImpl

public MaxPool2dImpl(@ByVal @Cast(value="torch::ExpandingArray<2>*")
                     LongPointer kernel_size)

MaxPool2dImpl

public MaxPool2dImpl(@Const @ByRef
                     MaxPool2dOptions options_)

MaxPool2dImpl

public MaxPool2dImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

MaxPool2dImpl

public MaxPool2dImpl(Module pointer)

Downcast constructor.

Method Detail

forward

@ByVal
public Tensor forward(@Const @ByRef
                             Tensor input)

forward_with_indices

@ByVal
public T_TensorTensor_T forward_with_indices(@Const @ByRef
                                                    Tensor input)

Returns the outputs and the indices of the max values. Useful for torch::nn::MaxUnpool2d later.