在PyTorch张量中的元素之间执行Delta函数

Question

我有一个一维pyTorch张量(dtype:int32(，想知道是否有办法对这个张量中的元素执行Dirac Delta函数，即：

f = tensor[1, 0, 0, 0, 1, 1, 0, 1, 1, 0, 1, 1]
f_after_dirac_delta = tensor[0, 1, 1, 0, 1, 0, 0, 1, 0, 0, 1]

感谢您提前提供的帮助！

编辑：正如@GirishDattatrayHegde所提到的，狄拉克三角洲一词具有误导性。正确的术语应该是克罗内克三角洲。我很抱歉。

Answer 1

如果我理解正确，您需要比较张量的连续元素。这应该有效：

import torch
f = torch.tensor([1, 0, 0, 0, 1, 1, 0, 1, 1, 0, 1, 1])
f_dirac = (f[1:] == f[:-1]).to(torch.long)