xma.utils.tensor¶

empty_like_contiguous(x: Tensor, dtype: dtype | None = None) → Tensor[source]¶

get_alignment(x: Tensor) → int[source]¶

get_num_elements_and_hidden_size(x: Tensor) → tuple[int][source]¶

zeros_like_contiguous(x: Tensor, dtype: dtype | None = None) → Tensor[source]¶