tensorflow1.15之间tf.py_function和tf.py_func的区别
实验总结来看:
tf.py_function()
输入:EagerTensor
返回:必须是Tensor
tf.py_func()
输入:ndarray
返回:必须是ndarray
实验代码
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import datasets
import numpy as np
print(tf.__version__)
def get_batch_.
原创
2020-10-13 11:19:29 ·
862 阅读 ·
2 评论