Function _pad_inf

eval_code/recons/models/moge/utils/alignment.py:40–41 · view source on GitHub ↗

(x_: torch.Tensor)

Source from the content-addressed store, hash-verified

38
39
40	def _pad_inf(x_: torch.Tensor):
41	return torch.cat([torch.full_like(x_[..., :1], -torch.inf), x_, torch.full_like(x_[..., :1], torch.inf)], dim=-1)
42
43
44	def _pad_cumsum(cumsum: torch.Tensor):

alignFunction · 0.70

no outgoing calls

no test coverage detected