昇腾社区首页
中文
注册

def decode

函数功能

视频帧解码函数。

函数原型

def decode(self, z, num_frames) -> torch.Tensor:

参数说明

参数名

输入/输出

类型

说明

z

输入

torch.Tensor

视频帧张量,必须为5维输入。

num_frames

输入

int

视频总帧数,当前仅支持32或128。

返回值说明

返回解码后的视频。