@@ -45,25 +45,26 @@ def compute_ect(
4545 other implemetations is most likely faster at the cost of less flexibility.
4646
4747 Args:
48- x:
49- The point cloud of shape [BxN,D] where B is the number of point clouds,
50- N is the number of points and D is the ambient dimension.
51- simplices:
52- Contains, as _ordered_ set of arguments, the index tensors for the
53- simplicial complex in ascending order. See examples.
54- v:
55- The tensor of directions of shape [D,N], where D is the ambient
56- dimension and N is the number of directions.
57- radius:
58- Radius of the interval to discretize the ECT into. (Is irrelevant for
59- this experiment.)
60- resolution:
61- Number of steps to divide the lin interval into.
62- scale:
63- The multipicative factor for the sigmoid function.
64- index:
65- Tensor of integers batching the points in their respective batch.
66- The index tensor is assumed to start at 0, otherwise fails.
48+ x:
49+ The point cloud of shape [BxN,D] where B is the number of point clouds,
50+ N is the number of points and D is the ambient dimension.
51+ simplices:
52+ Contains, as _ordered_ set of arguments, the index tensors for the
53+ simplicial complex in ascending order. See examples.
54+ v:
55+ The tensor of directions of shape [D,N], where D is the ambient
56+ dimension and N is the number of directions.
57+ radius:
58+ Radius of the interval to discretize the ECT into. (Is irrelevant for
59+ this experiment.)
60+ resolution:
61+ Number of steps to divide the lin interval into.
62+ scale:
63+ The multipicative factor for the sigmoid function.
64+ index:
65+ Tensor of integers batching the points in their respective batch.
66+ The index tensor is assumed to start at 0, otherwise fails.
67+
6768 Returns:
6869 The ECT of the point cloud of shape [B,N,R] where B is the number of
6970 point clouds (thus ECT's), N is the number of direction and R is the
0 commit comments