zeros_like関数【torch】
与えたテンソルと同じサイズのゼロ要素からなるテンソルを生成します。
クラスメソッドです。
code:zero_like1.py
import torch as pt
x = pt.tensor(1, 2, 3], [4, 5, 6)
y = pt.zeros_like(x)
print(type(y))
print(y.shape)
print(y)
https://scrapbox.io/files/650668ea2165bf001c59894e.png