pytorch版本实现

[zcswdt]

您好，我看到您这是tensorflow版本实现的，请问有pythorch版本实现吗？谢谢