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
]