生成一个m*n的tensor数组
时间: 2023-05-12 18:03:23 浏览: 121
Pytorch生成随机数Tensor的方法汇总
可以使用Python中的NumPy库来生成一个m*n的tensor数组,代码如下:
import numpy as np
m = 3
n = 4
tensor_array = np.zeros((m, n))
print(tensor_array)
这段代码将生成一个3行4列的tensor数组,并将其打印出来。你可以根据需要修改m和n的值来生成不同大小的数组。
阅读全文