tensorflow与pytorch张量互转

Posted 2023-02-16 AI 菌

1、tensorflow张量转pytorch张量

• tensorflow(Tensor)–>numpy.ndarray–>pytorch(Tensor)

import torch
import tensorflow as tf

tf_tensor = tf.constant([1,2,3])
with tf.compat.v1.Session().as_default():
	np_array = tf_tensor.eval()
torch_tensor = torch.from_numpy(np_array )

2、pytorch张量转tensorflow张量

• pytorch(Tensor)–>numpy.ndarray–>tensorflow(Tensor)

import torch
import tensorflow as tf

torch_tensor = torch.ones(100)
np_tensor = torch_tensor.numpy()
tf_tensor = tf.convert_to_tensor(np_tensor)