torch.nn.functional.softmin¶
- torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None)[source][source]¶
Apply a softmin function.
Note that . See softmax definition for mathematical formula.
See
Softmin
for more details.- Parameters
input (Tensor) – input
dim (int) – A dimension along which softmin will be computed (so every slice along dim will sum to 1).
dtype (
torch.dtype
, optional) – the desired data type of returned tensor. If specified, the input tensor is casted todtype
before the operation is performed. This is useful for preventing data type overflows. Default: None.
- Return type