Searched defs:TensorSequenceType (Results 1 – 3 of 3) sorted by relevance
2717 def cat(tensors: TensorSequenceType, dim: int = 0) -> TensorLikeType:2837 def column_stack(tensors: TensorSequenceType) -> TensorLikeType:2953 def dstack(tensors: TensorSequenceType) -> TensorLikeType:3872 def _check_stack_inputs(tensors: TensorSequenceType) -> None:3883 def stack(tensors: TensorSequenceType, dim: int = 0) -> TensorLikeType:3920 def hstack(tensors: TensorSequenceType) -> TensorLikeType:3930 def vstack(tensors: TensorSequenceType) -> TensorLikeType:6426 scalarType: torch.dtype, obj: Union[TensorOrNumberLikeType, TensorSequenceType] argument
93 TensorSequenceType: TypeAlias = Union[List[TensorLikeType], Tuple[TensorLikeType, ...]] variable
3708 def _compute_weight_precision(weights: TensorSequenceType) -> Tensor:3987 def _upsample_cubic_interp1d(coeffs: TensorSequenceType, ts: Tensor) -> Tensor: