nip.utils.torch.Squeeze#
- class nip.utils.torch.Squeeze(dim: int = -1)[source]#
Squeeze a dimension.
- Parameters:
dim (int, default=-1) – The dimension to squeeze.
Methods Summary
__init__
([dim])Initialize internal Module state, shared by both nn.Module and ScriptModule.
forward
(x)Squeeze the input tensor.
Attributes
T_destination
call_super_init
dump_patches
training
Methods