类 TensorFormat

• 定义在 文件 torch_tensorrt.h

类文档

class TensorFormat

TensorFormat 是一个枚举类，它定义用于存储张量数据的内存布局。

公共类型

enum Value

支持 TensorFormat 类的底层枚举类

如果您需要使用 TensorFormat 类本身，请使用此枚举进行接口，而不是正常实例化

例如。 torch_tensorrt::TensorFormat type = TensorFormat::kContiguous;

值

enumerator kContiguous

连续/NCHW/线性。

enumerator kChannelsLast

通道最后/NHWC。

enumerator kUnknown

哨兵值。

公共函数

TensorFormat() = default

构造一个新的 TensorFormat 对象。

inline constexpr TensorFormat(Value t)

从枚举构造 TensorFormat 构造函数。

TORCHTRT_API TensorFormat(at::MemoryFormat t)

从火炬类型枚举构造一个新的 TensorFormat 对象。

参数

t –

inline operator Value() const

获取 TensorFormat 对象的枚举值。

返回

值

explicit operator bool() = delete

inline constexpr bool operator==(TensorFormat other) const

用于 TensorFormat 的比较运算符。

参数

other –

返回

true

false

inline constexpr bool operator==(TensorFormat::Value other) const

用于 TensorFormat 的比较运算符。

参数

other –

返回

true

false

inline constexpr bool operator!=(TensorFormat other) const

用于 TensorFormat 的比较运算符。

参数

other –

返回

true

false

inline constexpr bool operator!=(TensorFormat::Value other) const

用于 TensorFormat 的比较运算符。

参数

other –

返回

true

false