pytorch_pfn_extras.nn.modules.ensure_shape#
Functions
Checks the shape and type of a tensor. |
Classes
Module to check the shape of a tensor. |
Site Navigation
Section Navigation
Functions
Checks the shape and type of a tensor. |
Classes
Module to check the shape of a tensor. |