KeyedJaggedTensor.from_jt_dict（TorchRec）
此接口为TorchRec开源接口，非Rec SDK Torch对外接口。此章节介绍使用Rec SDK Torch时调用的TorchRec接口支持的参数范围。
功能描述
通过JaggedTensor，创建KeyedJaggedTensor。
函数原型
1 | def from_jt_dict(jt_dict: Dict[str, JaggedTensor]) -> "KeyedJaggedTensor" |
参数说明
参数名 |
类型 |
可选/必选 |
说明 |
|---|---|---|---|
jt_dict |
Dict[str, JaggedTensor] |
必选 |
特征名称和对应的JaggedTensor组成的字典。长度不能为0。其中JaggedTensor的取值范围参考JaggedTensor（TorchRec）。 |
返回值说明
- 成功：返回KeyedJaggedTensor。
- 失败：抛出异常。
使用示例
1 2 3 | from torchrec import KeyedJaggedTensor, JaggedTensor jt = JaggedTensor(values=[1, 3, 4], lengths=[1, 1, 1]) KeyedJaggedTensor.from_jt_dict({"feat0": jt}) |
参考资源
接口调用流程及示例，参见迁移与训练。
父主题： 数据接口