Rate this Page
★★★★★
torch.trunc#
- torch.trunc(input,*,out=None)→Tensor#
Returns a new tensor with the truncated integer values ofthe elements of
input.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.])
On this page