ShapeLayer¶

class ShapeLayer(shapes, use_cpp=False)[source]¶

Bases: sphinx.ext.autodoc.importer._MockObject

Wrapper to compine Python and C++ Implementation under Single API

forward(shape_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843aa7dd8>)[source]¶

Forwards parameters to Python or C++ Implementation

Parameters:	shape_params (`torch.Tensor`) – parameters for shape ensembling
Returns:	Ensempled Shape
Return type:	`torch.Tensor`

num_params¶

Property to access these layer’s parameters

Returns:	number of parameters
Return type:	int

_ShapeLayerPy¶

class _ShapeLayerPy(shapes)[source]¶

Bases: sphinx.ext.autodoc.importer._MockObject

Python Implementation of Shape Layer

forward(shape_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843aa7e80>)[source]¶

Ensemble shape from parameters

Parameters:	shape_params (`torch.Tensor`) – shape parameters
Returns:	ensembled shape
Return type:	`torch.Tensor`

num_params¶

Property to access these layer’s parameters

Returns:	number of parameters
Return type:	int

class _ShapeLayerCpp(shapes, verbose=True)[source]¶

Bases: sphinx.ext.autodoc.importer._MockObject

C++ Implementation of Shape Layer

forward(shape_params: <sphinx.ext.autodoc.importer._MockObject object at 0x7f5843aa7f28>)[source]¶

Ensemble shape from parameters

Parameters:	shape_params (`torch.Tensor`) – shape parameters
Returns:	ensembled shape
Return type:	`torch.Tensor`

num_params¶

Property to access these layer’s parameters

Returns:	number of parameters
Return type:	int