org.bytedeco.pytorch

Class PixelShuffleImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.PixelShuffleImplCloneable

org.bytedeco.pytorch.PixelShuffleImpl

All Implemented Interfaces:

AutoCloseable

@Namespace(value="torch::nn")
 @NoOffset
 @Properties(inherit=torch.class)
public class PixelShuffleImpl
extends PixelShuffleImplCloneable

Rearranges elements in a tensor of shape :math:(*, C \times r^2, H, W) to a tensor of shape :math:(*, C, H \times r, W \times r), where r is an upscale factor. See https://pytorch.org/docs/master/nn.html#torch.nn.PixelShuffle to learn about the exact behavior of this module. See the documentation for torch::nn::PixelShuffleOptions class to learn what constructor arguments are supported for this module. Example:

  PixelShuffle model(PixelShuffleOptions(5));
  

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
PixelShuffleImpl(Module pointer) Downcast constructor.
PixelShuffleImpl(PixelShuffleOptions options_)
PixelShuffleImpl(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input)
PixelShuffleOptions	options() The options with which this Module was constructed.
PixelShuffleImpl	options(PixelShuffleOptions setter)
void	pretty_print(Pointer stream) Pretty prints the PixelShuffle module into the given stream.
void	reset() reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Methods inherited from class org.bytedeco.pytorch.PixelShuffleImplCloneable

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

PixelShuffleImpl

public PixelShuffleImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

PixelShuffleImpl

public PixelShuffleImpl(Module pointer)

Downcast constructor.

PixelShuffleImpl

public PixelShuffleImpl(@Const @ByRef
                        PixelShuffleOptions options_)

Method Detail

pretty_print

public void pretty_print(@Cast(value="std::ostream*") @ByRef
                         Pointer stream)

Pretty prints the PixelShuffle module into the given stream.

Overrides:

pretty_print in class Module

forward

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

reset

public void reset()

Description copied from class: PixelShuffleImplCloneable

reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Overrides:

reset in class PixelShuffleImplCloneable

options

@ByRef
public PixelShuffleOptions options()

The options with which this Module was constructed.

options

public PixelShuffleImpl options(PixelShuffleOptions setter)