Class TensorFormat

• 定义于 文件 torch_tensorrt.h

类文档

class TensorFormat

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

公共类型

enum Value

底层的枚举类，用于支持 TensorFormat 类

如果你需要使用 TensorFormat 类本身，可以通过此枚举进行接口调用，而非正常实例化。

例如：torch_tensorrt::TensorFormat 类型 = 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)

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

参数

t –

inline operator Value() const

获取 TensorFormat 对象的枚举值。

返回

值

explicit operator bool() = delete

inline constexpr bool operator==(TensorFormat other) const

TensorFormat 的比较运算符。

参数

其他 –

返回

真

返回

假

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

TensorFormat 的比较运算符。

参数

其他 –

返回

真

返回

假

inline constexpr bool operator!=(TensorFormat other) const

TensorFormat 的比较运算符。

参数

其他 –

返回

真

返回

假

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

TensorFormat 的比较运算符。

参数

其他 –

返回

真

返回

假