Rate this Page

torch.trunc#

torch.trunc(input,*,out=None)Tensor#

Returns a new tensor with the truncated integer values ofthe elements ofinput.

For integer inputs, follows the array-api convention of returning acopy of the input tensor.

Parameters

input (Tensor) – the input tensor.

Keyword Arguments

out (Tensor,optional) – the output tensor.

Example:

>>>a=torch.randn(4)>>>atensor([ 3.4742,  0.5466, -0.8008, -0.9079])>>>torch.trunc(a)tensor([ 3.,  0., -0., -0.])