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(a_n)
# tensor --> ndarray
sess = tf.Session()
a_n = sess.run(a_tf)
# ==> Torch 版本
# ndarray --> tensor
a_torch = torch.from_numpy(a_n)
# tensor --> ndarray
a_n = a_torch.array()