Rate this Page
★★★★★

torch.nn.functional.softmin#

torch.nn.functional.softmin(input,dim=None,_stacklevel=3,dtype=None)[source]#

Apply a softmin function.

Note thatSoftmin(x)=Softmax(−x)\text{Softmin}(x) = \text{Softmax}(-x). See softmax definition for mathematical formula.

SeeSoftmin for more details.

Parameters

  • input (Tensor) – input

  • dim (int) – A dimension along which softmin will be computed (so every slicealong 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 operationis performed. This is useful for preventing data type overflows. Default: None.

Return type

Tensor