我是 pytorch 的新手，一直在尝试转换一些代码。找不到此特定功能。它存在于 tensorflow 中吗？
numpy deep-learning keras tensorflow pytorch
deep-learning ×1
keras ×1
numpy ×1
pytorch ×1
tensorflow ×1