torch Namespace Reference

Functions
list[list[float]]	tensor (list[list[float]] data)
list[list[float]]	mm (list[list[float]] A, list[list[float]] B)
list[list[float]]	matmul (list[list[float]] A, list[list[float]] B)
list[list[float]]	cat (list[list[list[float]]] tensors, int dim)
list[list[list[float]]]	split (list[list[float]] tensor, list[int] sizes, int dim)
bool	allclose (list[list[float]] a, list[list[float]] b, float rtol=1e-05, float atol=1e-08)

Function Documentation

allclose()

bool torch.allclose	(	list[list[float]]	a,
		list[list[float]]	b,
		float	rtol = 1e-05,
		float	atol = 1e-08 )

cat()

list[list[float]] torch.cat	(	list[list[list[float]]]	tensors,
		int	dim )

matmul()

list[list[float]] torch.matmul	(	list[list[float]]	A,
		list[list[float]]	B )

mm()

list[list[float]] torch.mm	(	list[list[float]]	A,
		list[list[float]]	B )

split()

list[list[list[float]]] torch.split	(	list[list[float]]	tensor,
		list[int]	sizes,
		int	dim )

tensor()

list[list[float]] torch.tensor

(

list[list[float]]

data

)