pytorch_pfn_extras.nn

Functions

pytorch_pfn_extras.nn.ensure(tensor[, ...])

Checks the shape and type of a tensor.

Classes

pytorch_pfn_extras.nn.Ensure(*[, shape, ...])	Module to check the shape of a tensor.
pytorch_pfn_extras.nn.ExtendedSequential()	Sequential module with extended features from chainer.
pytorch_pfn_extras.nn.LazyBatchNorm1d(...)	BatchNorm1d module with lazy weight initialization.
pytorch_pfn_extras.nn.LazyBatchNorm2d(...)	BatchNorm2d module with lazy weight initialization.
pytorch_pfn_extras.nn.LazyBatchNorm3d(...)	BatchNorm3d module with lazy weight initialization.
pytorch_pfn_extras.nn.LazyConv1d(...)	Conv1d module with lazy weight initialization.
pytorch_pfn_extras.nn.LazyConv2d(...)	Conv2d module with lazy weight initialization.
pytorch_pfn_extras.nn.LazyConv3d(...)	Conv3d module with lazy weight initialization.
pytorch_pfn_extras.nn.LazyLinear(...)	Linear module with lazy weight initialization.

Modules

pytorch_pfn_extras.nn.modules
pytorch_pfn_extras.nn.parallel