编程语言
首页 > 编程语言> > 从 Numpy+Pytorch 到 TensorFlow JS:总结和常用平替整理

从 Numpy+Pytorch 到 TensorFlow JS:总结和常用平替整理

作者:互联网

如何拥有较为平滑的移植体验?

一些碎碎念

常用平替整理

将张量转换为数组

tensor = torch.tensor([1,2,3])
np_array = tensor.numpy()
// 方式一:arraySync()
let tensor = tf.tensor1d([1,2,3]);
let array = tensor.arraySync();
console.log(array); // [1,2,3]

// 方式二:在async函数体内操作
async function fun() {
    let tensor = tf.tensor1d([1,2,3]);
    let array = await tensor.array();
    console.log(array); // [1,2,3]
}
fun();

// 注意,下面的写法是不行的,因为async函数的返回值是Promise对象
array = async function (){
    return await tensor.array();
}();
console.log(array); // Promise object

// 方式三:用then取出async函数返回Promise对象中的值
let a
(async function() {
    let array = await tensor.array(); 
    return array
})().then(data => {a = data;})
console.log(a); // [1,2,3]

访问张量中的元素

tensor = torch.tensor([1,2,3])
print(tensor[0])
print(tensor[-1])

标签:Numpy,Pytorch,TensorFlow,语言,框架,JavaScript,浏览器,关键词
来源: