类 TensorFormat¶
类文档¶
-
class TensorFormat¶
TensorFormat 是一个枚举类，用于定义存储张量数据的内存布局。
公共类型
-
enum Value¶
支持TensorFormat类的基础枚举类
在需要使用TensorFormat类本身的情况下，使用此枚举与普通实例化进行接口
例如，torch_tensorrt::TensorFormat 类型 = TensorFormat::kContiguous;
值：
-
enumerator kContiguous¶
连续的 / NCHW / 线性的。
-
enumerator kChannelsLast¶
通道最后 / NHWC。
-
enumerator kUnknown¶
哨兵值。
-
enumerator kContiguous¶
公共函数
-
TensorFormat() = default¶
构造一个新的TensorFormat对象。
-
inline constexpr TensorFormat(值 t)¶
TensorFormat 从枚举构造函数。
-
TORCHTRT_API TensorFormat(at::MemoryFormat t)¶
从torch类型枚举中构造一个新的TensorFormat对象。
- Parameters
t –
-
inline operator 值() const¶
获取TensorFormat对象的枚举值。
- Returns
值
-
explicit operator bool() = delete¶
-
inline constexpr bool operator==(TensorFormat other) const¶
用于TensorFormat的比较运算符。
- Parameters
其他 –
- Returns
真
- Returns
假
-
inline constexpr bool operator==(TensorFormat::值 other) const¶
TensorFormat 的比较运算符。
- Parameters
其他 –
- Returns
真
- Returns
假
-
inline constexpr bool operator!=(TensorFormat other) const¶
TensorFormat 的比较运算符。
- Parameters
其他 –
- Returns
真
- Returns
假
-
inline constexpr bool operator!=(TensorFormat::值 other) const¶
TensorFormat 的比较运算符。
- Parameters
其他 –
- Returns
真
- Returns
假
-
enum Value¶
