Method from_tensor

exir/tensor_layout.py:27–39

(cls, tensor: torch.Tensor)

    @classmethod
    def from_tensor(cls, tensor: torch.Tensor) -> "TensorLayout":
        if not (
            tensor.is_contiguous(memory_format=torch.contiguous_format)
            or tensor.is_contiguous(memory_format=torch.channels_last)
        ):
            raise ValueError(
                "Tensor is not contiguous. Please call .contiguous() before creating the TensorLayout."
            )
        return TensorLayout(
            scalar_type=scalar_type_enum(tensor.dtype),
            sizes=list(tensor.shape),
            dim_order=list(dim_order_from_stride(tensor.stride())),
        )