deepfold.utils.geometry.vector.Vec3Array

class deepfold.utils.geometry.vector.Vec3Array(x: 'torch.Tensor', y: 'torch.Tensor', z: 'torch.Tensor')[source]
__init__(x: Tensor, y: Tensor, z: Tensor) None

Methods

__init__(x, y, z)

cat(vecs, dim)

clone()

cross(other)

Compute cross product between 'self' and 'other'.

dot(other)

Compute dot product between 'self' and 'other'.

from_array(tensor)

map_tensor_fn(fn)

norm([epsilon])

Compute Norm of Vec3Array, clipped to epsilon.

norm2()

normalized([epsilon])

Return unit vector with optional clipping.

reshape(new_shape)

sum(dim)

to_tensor()

unsqueeze(dim)

zeros(shape[, device])

Return Vec3Array corresponding to zeros of given shape.

Attributes

shape

x

y

z