TensorLike

lightkit.data.TensorLike

Type annotation for functions accepting any kind of tensor data as input. Consider using this annotation if your methods in an estimator derived from lightkit.BaseEstimator work on tensors.

alias of Union[numpy.ndarray[Any, numpy.dtype[numpy.float32]], torch.Tensor]