Computes the n-th forward difference along the given dimension.
The first-order differences are given by out[i] = input[i + 1] - input[i]. Higher-order differences are calculated by using torch.diff()
recursively.
input (Tensor) – the tensor to compute the differences on
n (int, optional) – the number of times to recursively compute the difference
dim (int, optional) – the dimension to compute the difference along. Default is the last dimension.
prepend (Tensor, optional) – values to prepend or append to input
along dim
before computing the difference. Their dimensions must be equivalent to that of input, and their shapes must match input’s shape except on dim
.
append (Tensor, optional) – values to prepend or append to input
along dim
before computing the difference. Their dimensions must be equivalent to that of input, and their shapes must match input’s shape except on dim
.
out (Tensor, optional) – the output tensor.
Example:
>>> a = torch.tensor([1, 3, 2]) >>> torch.diff(a) tensor([ 2, -1]) >>> b = torch.tensor([4, 5]) >>> torch.diff(a, append=b) tensor([ 2, -1, 2, 1]) >>> c = torch.tensor([[1, 2, 3], [3, 4, 5]]) >>> torch.diff(c, dim=0) tensor([[2, 2, 2]]) >>> torch.diff(c, dim=1) tensor([[1, 1], [1, 1]])
Access comprehensive developer documentation for PyTorch
View Docs TutorialsGet in-depth tutorials for beginners and advanced developers
View Tutorials ResourcesFind development resources and get your questions answered
View ResourcesRetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4