リストからのテンソルの生成

ここではリストや配列などのオブジェクトから生成する方法を学ぶ。

１次元のテンソル

code:tensor_list1.py

import torch as pt

print('１次元のテンソルの例')

x = pt.tensor(data) # リストからtensor型の変数xを生成

print('# tensor型')

print(x.shape) # 形状

print(type(x)) # 型の確認

print(x) # 値

２次元のテンソル

code:tensor_list2.py

import torch as pt

print('２次元のテンソルの例')

print('型 :', type(x))

print('値 :', x)

print('形状:', x.shape)

shape属性【torch】で返される形状についての情報はtorch.Size型というクラス型をもつ。これはタプルのように添字を使って各軸についてのサイズを取り出すことができる。

code:(続き).py

a = x.shape

code:tensor_list3.py

import torch as pt

print('x ----------------')

print(type(x))

print(x)

print('y ----------------')

print(type(y))

print(y)

code:(結果).py

x ----------------

<class 'torch.Tensor'>

y ----------------

<class 'torch.Tensor'>