KeyedJaggedTensorWithTimestamp

Function

This API is a class extended from KeyedJaggedTensor and is used to represent a KeyedJaggedTensor with timestamp information. This class adds the _timestamps attribute based on KeyedJaggedTensor to store the timestamp information corresponding to values. Used to calculate the time during feature elimination.

Prototype

1
def from_jt_dict(jt_dict: Dict[str, JaggedTensorWithTimestamp]) -> "KeyedJaggedTensorWithTimestamp"

Parameters

Parameter

Data Type

Mandatory/Optional

Description

jt_dict

Dict[str, JaggedTensorWithTimestamp]

Mandatory

Dictionary consisting of the feature name and the corresponding JaggedTensorWithTimestamp. The length cannot be 0. For details about the value range of JaggedTensorWithTimestamp, see JaggedTensor (TorchRec).