TensorFlow和Torch的对比可以参考 TensorFlow vs. Pytorch
在做网络处理时,经过预测处理之后的数据经常是numpy生成的数据格式,有时候和网络层交互时需要和tensor互转,以下提供几个tensor和ndarray之间互转的样例
import tensorflow as tf
import numpy as np
import torch
# 生成ndarray
a_n = np.arange(10)
# ==> TensorFlow 版本
# ndarray --> tensor
a_tf = tf.convert_tensor