torch.py File Reference

Namespaces
namespace	torch

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