linxin02
/

pixel_gen

Model card Files Files and versions

pixel_gen / src /models /layers /attention_op.py

linxin02's picture

Upload lx_gan project

cef8b68 verified about 2 months ago

History Blame Contribute Delete

113 Bytes

	import torch
	import torch.nn as nn

	from torch.nn.functional import scaled_dot_product_attention as attention