PyTorch
torch / torch
torch.trunc¶
-
torch.
trunc
(input, *, out=None) → Tensor¶ Returns a new tensor with the truncated integer values of the elements of
input
.- Parameters
input (Tensor) – the input tensor.
- Keyword Arguments
out (Tensor, optional) – the output tensor.
Example:
>>> a = torch.randn(4) >>> a tensor([ 3.4742, 0.5466, -0.8008, -0.9079]) >>> torch.trunc(a) tensor([ 3., 0., -0., -0.])
此页内容是否对您有帮助
感谢反馈!