配列からのテンソルの生成

関数tensor【NumPy】に配列を渡すことで、テンソルを生成することができる。

code:tensor_array1.py

import numpy as np

import torch as pt

l1 = np.array(1, 2, 3)

t1 = pt.tensor(l1)

print(l1)

print(t1)

構成要素のデータ型は配列のものが引き継がれる。引数dtypeを利用することで生成後の型を指定することができる。

code:tensor_array2.py

import numpy as np

import torch as pt

l1 = np.array(1, 2, 3)

t1 = pt.tensor(l1, dtype=pt.int)

t2 = pt.tensor(l1, dtype=pt.float)

print(l1)

print(t1)

print(t2)