Documentation for v0.1 version (#199)
* Adding Equations, solving typos * improve _code.rst * the team rst and restuctore index.rst * fixing errors --------- Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
This commit is contained in:
committed by
Nicola Demo
parent
3f9305d475
commit
8b7b61b3bd
@@ -3,8 +3,8 @@ import torch
|
||||
|
||||
def check_point(x, current_stride, dim):
|
||||
max_stride = current_stride + dim
|
||||
indeces = torch.logical_and(x[..., :-1] < max_stride,
|
||||
x[..., :-1] >= current_stride).all(dim=-1)
|
||||
indeces = torch.logical_and(x[..., :-1] < max_stride, x[..., :-1]
|
||||
>= current_stride).all(dim=-1)
|
||||
return indeces
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user